Philippe Tillet d11d2db6ee [PYTHON][EINSUM] Now handling reduction sizes that are not a multiple of ...

TK

2020-02-17 13:52:58 -05:00

..

_C

[python] more robust way to add triton includes to python package

2019-09-05 16:01:56 -04:00

ops

[PYTHON][EINSUM] Now handling reduction sizes that are not a multiple of

2020-02-17 13:52:58 -05:00

__init__.py

[python] now packaging include and libtriton in triton._C submodule

2019-09-05 15:37:00 -04:00

frameworks.py

[PYTHON] Removed dead code for alloc_empty and register_scalar

2019-10-30 10:37:30 -04:00

function.py

[GENERAL] Merged einsum feature branch. Various feature, performance

2020-01-20 12:42:48 -05:00

kernel.py

[PYTHON][EINSUM] Now handling reduction sizes that are not a multiple of

2020-02-17 13:52:58 -05:00

utils.py

[PYTHON][EXAMPLES] Added self-attention example using triton.ops.einsum

2020-01-21 16:45:04 -05:00