Philippe Tillet
d11d2db6ee
[PYTHON][EINSUM] Now handling reduction sizes that are not a multiple of
...
TK
2020-02-17 13:52:58 -05:00
Philippe Tillet
6a4d42c1b8
[PYTHON][CORE] Deprecating Tensorflow support
2020-02-10 04:20:33 -05:00
Philippe Tillet
5a3c30148e
[PYTHON][EXAMPLES] Changed shape of einsum examples
2020-02-06 13:57:30 -05:00
Philippe Tillet
3e92901bd5
[TRITON][PYTHON] Cleaned up API
2020-02-05 19:44:19 -05:00
Philippe Tillet
ce7a00674a
[PYTHON][EXAMPLES] Added self-attention example using triton.ops.einsum
2020-01-21 16:45:04 -05:00
Philippe Tillet
78b98fb7cf
[GENERAL] Cleaned polymorphic structure of layouts analysis pass
2020-01-21 11:38:39 -05:00
Philippe Tillet
fbf2a3f56f
[CODEGEN][TRANSFORM] some bug-fixes for FP32 einsum
2020-01-20 12:42:53 -05:00
Philippe Tillet
f278d9741a
[GENERAL] Merged einsum feature branch. Various feature, performance
...
improvements and bugfixes:
* Added preliminary support for extended Einstein summation in PyTriton
* Significant performance improvement on FP32 kernels containing matrix
multiplication
* Added re-coalescing pass for FP16 kernels containing matrix
multiplication
* Various bugfixes
2020-01-20 12:42:48 -05:00
Philippe Tillet
bf3dc63858
[PYTHON] Removed dead code for alloc_empty and register_scalar
2019-10-30 10:37:30 -04:00
Philippe Tillet
76651a065f
[PYTHON][EXAMPLES] Better einsum example
2019-10-29 12:56:58 -04:00
Philippe Tillet
943bf41b5c
[python] [op] added Triton NT einsum
2019-10-21 23:37:39 -04:00
Philippe Tillet
099918b3c0
[python] [ops] added skeleton for einsum op
2019-10-21 18:58:02 -04:00
Philippe Tillet
4b0c43bb7b
[python][example] added test for einsum
2019-10-21 17:13:12 -04:00
