Philippe Tillet
a099c6f7f3
[TRITON][LANG] Added support for bitcast
2020-02-09 20:11:13 -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
d65a94c768
[PYTHON][OPS] Added batch normalization op
2019-10-29 17:29:11 -04:00
Philippe Tillet
76adcb755a
[PYTHON][EXAMPLES] Tentative support for einsum with transpositions
2019-10-25 19:01:21 -04:00
Philippe Tillet
abe3fbb480
[test] [reduce] added test for 1D reduction
2019-10-20 01:01:53 -04:00
Philippe Tillet
d76c6bc3c7
Merge branch 'master' into auto-coalesce
2019-10-18 16:21:28 -04:00
Philippe Tillet
f4beb713ab
[test] added support for max, min reduction and made it easy to add more
2019-09-12 16:11:57 -04:00
Philippe Tillet
04a0fbd8e3
[tests] basic test for reduction in python passes
2019-09-11 17:35:56 -04:00
Philippe Tillet
0c41bade07
[codegen] basic recoalescing working
2019-09-10 23:25:47 -04:00
Philippe Tillet
a842d337c5
[general] various cleaning and bugfix:
...
* added copy1d and copy2d benchmark
* fixed issue in reassociation pass
2019-09-02 23:00:49 -04:00
Philippe Tillet
37cbcfabd0
[examples] back to 96 TFLOPS on V100
2019-08-26 22:49:14 -07:00
Philippe Tillet
44eb3891ae
[lang] added support for restrict; added macros for attributes
2019-08-23 20:29:12 -07:00
Philippe Tillet
8c6bac49d1
[lang][codegen] added basic attribute support
2019-08-23 19:49:06 -07:00
Philippe Tillet
cb04ec0b3b
some more cleaning
2019-08-23 19:22:38 -07:00
Philippe Tillet
a110a7e8cf
[ir] changed type of tile shapes from constant_int* to int
2019-08-23 17:49:21 -07:00
Philippe Tillet
f98b0b8e2a
[general] deleted the old compiler frontend
2019-08-23 17:28:02 -07:00