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
f4bbbbe5e4
[PYTHON][OPS] Bugfix in conv fprop
2019-11-01 00:43:02 -04:00
Philippe Tillet
d65a94c768
[PYTHON][OPS] Added batch normalization op
2019-10-29 17:29:11 -04:00
Philippe Tillet
e9c787ef05
[PYTHON][EINSUM] Added support for FP16
2019-10-28 14:07:17 -04:00
Philippe Tillet
e11557855f
[PYTHON] [OPS] Added einsum implementation
2019-10-26 22:14:50 -04:00
Philippe Tillet
655f43fb5b
more work
2019-10-26 15:10:19 -04:00
Philippe Tillet
b615af2e7e
[codegen] [generator] fixed issue when tile size is 1 along one or more
...
dimensions
2019-10-25 14:22:28 -04:00
Philippe Tillet
abe3fbb480
[test] [reduce] added test for 1D reduction
2019-10-20 01:01:53 -04:00
Philippe Tillet
23db500edf
[tests] [common] added reduce.h to common headers
2019-10-19 16:53:48 -04:00
Philippe Tillet
a76efd326d
[selection] [codegen] added reduction
2019-10-19 14:47:16 -04:00
Philippe Tillet
d76c6bc3c7
Merge branch 'master' into auto-coalesce
2019-10-18 16:21:28 -04:00
Philippe Tillet
cf4fbfefee
[codegen] [selection] no longer using llvm::IRBuilder<>::Insert()
2019-10-17 13:12:37 -04:00
Philippe Tillet
be25e954f6
[codegen] [selection] merged selection into generator visit
2019-10-17 12:55:37 -04:00
Philippe Tillet
f4f70db234
[codegen] [selection] re-arranged file structure
2019-10-17 12:31:26 -04:00
