6 Commits

Author	SHA1	Message	Date
Philippe Tillet	7c09ff80eb	[CORE] Fixed several issues that arose in the development of the ... torch-blocksparse package: * Now using warp shuffle in reductions when possible * Various bugfixes in layout inference * Added INFINITY, exponential and select * Better error messages for unimplemented constructs	2020-03-31 18:57:28 -04:00
Philippe Tillet	382ca2c745	[CODEGEN][ANALYSIS] cleaning: moving towards better polymorphism for ... tile layouts	2020-01-20 12:43:04 -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	655f43fb5b	more work	2019-10-26 15:10:19 -04:00
Philippe Tillet	f4f70db234	[codegen] [selection] re-arranged file structure	2019-10-17 12:31:26 -04:00