418 Commits

Author SHA1 Message Date
Philippe Tillet
268894a5ce [PYTHON] Merged blocksparse branch:
* Example for blocksparse matrix multiplication
* Simplified Triton kernel API
* Revived auto-tuning in einsum
2021-07-27 12:38:48 -07:00
Philippe Tillet
ea37ba5d35 [PYTHON][OPS] Fixed typo in einsum 2021-07-27 12:38:48 -07:00
Philippe Tillet
926acc2e28 [TRITON][NN][CONV] Renamed input -> x to not modify built-in functions 2021-07-27 12:38:48 -07:00
Philippe Tillet
420e36a038 [PYTHON][NN][CONV] Fixed typo in dx computation 2021-07-27 12:38:48 -07:00
Philippe Tillet
01154f24db [PYTHON][SETUP] Removed obsolete debug print() 2021-07-27 12:38:48 -07:00
Philippe Tillet
ecb0d81b2d [PYTHON] Added missing files for nn submodule 2021-07-27 12:38:48 -07:00
Philippe Tillet
3d769b57e2 [PYTHON] Better packaging 2021-07-27 12:38:48 -07:00
Philippe Tillet
dfb844bf41 [GENERAL] Improved caching mechanism:
* Now computing hash in libtriton
* Now only compiling a single pytorch hook per function signature
2021-07-27 12:38:48 -07:00
Philippe Tillet
30f77e9ec5 [PYTHON][OPS][EINSUM] Now throwing error for automatic differentiation
of extended einsum
2021-07-27 12:38:48 -07:00
Philippe Tillet
4e50ef4076 [PYTHON][OP][EINSUM] simplified API 2021-07-27 12:38:48 -07:00
Philippe Tillet
26fd884d96 [PYTHON][OPS][EINSUM] Added support for inner tensor strides 2021-07-27 12:38:48 -07:00
Philippe Tillet
4181f9f2af [CODEGEN][TRANSFORM][PEEPHOLE] Fixed bug in *1 multiplication 2021-07-27 12:38:48 -07:00
Philippe Tillet
9e54a03006 [PYTHON][EXAMPLES] Removed obsolete files 2021-07-27 12:38:48 -07:00
Philippe Tillet
3816f2f259 [PYTHON][EINSUM] Now handling reduction sizes that are not a multiple of
TK
2021-07-27 12:38:48 -07:00
Philippe Tillet
fa4ec7ea65 [PYTHON][OPS][EINSUM] Added support for masked accumulator 2021-07-27 12:38:48 -07:00
Philippe Tillet
404dd18333 [PYTHON][CORE] Deprecating Tensorflow support 2021-07-27 12:38:48 -07:00
Philippe Tillet
558422c18a [PYTHON][EXAMPLES] Changed shape of einsum examples 2021-07-27 12:38:48 -07:00
Philippe Tillet
6d7cf35123 History prior to this date belonged to the now deprecated ISAAC project, and was deleted to save space 2021-07-27 12:38:38 -07:00