9 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	2fcf5cec5b	[TRITON][CODEGEN] Fixed flawed assert()	2020-01-24 15:25:00 -05:00
Philippe Tillet	78b98fb7cf	[GENERAL] Cleaned polymorphic structure of layouts analysis pass	2020-01-21 11:38:39 -05:00
Philippe Tillet	382ca2c745	[CODEGEN][ANALYSIS] cleaning: moving towards better polymorphism for ... tile layouts	2020-01-20 12:43:04 -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	cfde3dd766	[codegen] [layout] fixed padding issue for row-major HMMA	2019-10-18 13:42:15 -04:00
Philippe Tillet	b43454c9b7	[codegen] [membar] view do not write to shared memory	2019-10-17 22:38:41 -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