Commit Graph

142 Commits

Author SHA1 Message Date
Philippe Tillet
af080740f2 [GENERAL] Merged v1.0alpha into master. Added features are:
- A100 support via mma.16816
- Thread swizzling for conflict-free shared memory accesses without
padding
- Complete overhaul of the LLVM code generation in
codegen/selection/generator.cc to remove overengineering
- Added debugging capabilities in the Python binding
- Compilation error for kernels that spill
2021-01-11 19:23:24 -05:00
Philippe Tillet
0cf2d22ffc [PYTHON] Added TRITON_DEBUG_MODE which reallocates input tensors outside of the pytorch memory pool to spot out-of-bounds accesses more easily 2020-12-26 15:15:46 -05:00
Philippe Tillet
d913cbd916 [VERSION] 0.2.3 -> 0.3.0 2020-12-08 23:34:55 -05:00
Philippe Tillet
22fc1cef16 [PYTHON] Context switching logic moved to PyTorch 2020-11-26 22:46:27 -05:00
Philippe Tillet
7710e048f4 [DRIVER] Simplified Driver API by substantially removing reliance on driver::context 2020-11-26 00:38:25 -05:00
Philippe Tillet
fa066b531c [PYTHON] Compiling Triton in Release mode now... 2020-11-13 01:44:52 -05:00
Philippe Tillet
51025ca2ad [DRIVER] Improved performance of Host driver code 2020-11-12 02:11:45 -05:00
Philippe Tillet
6c5284ed3b [GENERAL] Various bugfixes 2020-11-11 14:44:56 -05:00
Philippe Tillet
a2d54b5ad3 [General] LLVM-9 -> LLVM-10 2020-11-07 22:46:18 -05:00
Philippe Tillet
81000db9e9 [PYTHON] Added option to show PTX source code in Python 2020-11-07 02:55:48 -05:00
Philippe Tillet
c1920cbabb [PYTHON] Fixed torch ABI issue 2020-11-05 12:37:00 -05:00
Philippe Tillet
ced0f5f944 [GENERAL] Fix compatibility issue with older Torch versions 2020-11-03 16:02:02 -05:00
Philippe Tillet
e2c1ac8d24 [LANG] Added log intrinsic 2020-11-03 15:50:11 -05:00
Philippe Tillet
37ee888d88 [PYTHON] Cleaning C++ bindings 2020-11-02 15:06:08 -05:00
Philippe Tillet
9be1d5afc2 [GENERAL] Various improvements:
* Sparse einsum in triton.ops.einsum
* Hacky support for fixed-tile-size atomic-add
* Various bugfixes in parser
2020-10-25 12:16:40 -07:00
Philippe Tillet
0cbee3ec56 [CODEGEN] More work on the CPU backend 2020-09-14 10:59:40 -04:00
Philippe Tillet
526a3468e0 [Version] Now version 0.2.3 2020-09-03 14:36:30 -04:00
Philippe Tillet
010f85e91d [VERSION] Now version 0.2.2 2020-08-31 15:16:09 -04:00
Philippe Tillet
3195aca452 [PYTHON] Added credits to Scott Gray for the idea used in launch.cc 2020-08-12 19:39:57 -04:00
Philippe Tillet
30ac1359b9 [RUNTIME] Lower-level interface for executing functions 2020-08-12 18:33:35 -04:00
Philippe Tillet
2d6484482f [CODEGEN][ANALYSIS] Fixed issue in layout inference 2020-08-10 11:53:11 -04:00
Philippe Tillet
bd2067606c [EXAMPLES] Improved mat_mul example 2020-08-06 17:29:52 -04:00
Philippe Tillet
f01bdd6207 [EXAMPLES] Added conv2d example 2020-08-06 17:29:52 -04:00
Philippe Tillet
ec923cfc5e [PACKAGING] Now version 0.2.1 2020-07-29 14:07:28 -04:00
Philippe Tillet
3683b38963 [PACKAGING] Now version 0.2.0 2020-07-14 19:50:39 -04:00
Philippe Tillet
cc7c77246b [EXAMPLES][TUTORIAL] Changed to new triton.kernel API 2020-07-08 13:39:19 -04:00
Philippe Tillet
baabaf0d58 [TRITON][PYTHON] Reverted back to distutils 2020-06-24 16:39:49 -04:00
Philippe Tillet
8e3ec0556f [TRITON][KERNEL] Fixed issue for concurrent compilation of torch
extensions
2020-06-24 15:47:43 -04:00
Philippe Tillet
b48460f51c [PACKAGING] Now version 0.1.3 2020-06-05 12:01:09 -04:00
Philippe Tillet
18a4cefec7 [CORE] Auto-tuning now copies scalar buffers. Still needs to copy all buffers that are both read from and written to. 2020-05-15 23:21:42 -04:00
Philippe Tillet
735215d9c4 [PYTHON] Added support for FP16 scalar kernel arguments 2020-05-15 16:37:25 -04:00
Philippe Tillet
f8614b2490 [PACKAGING] Now version 0.1.2 2020-05-12 20:50:52 -04:00
Philippe Tillet
8f9233e546 [LANG] Fixed undefined behavior in replace_all_uses_with() 2020-05-12 20:31:10 -04:00
Philippe Tillet
142a1dd42b [PYTHON] Restored compatibility with powerpc 2020-05-09 23:46:25 -04:00
Philippe Tillet
f88a7a0027 [PYTHON] Removed -std=gnu++11 in extra_cflags 2020-05-09 22:57:40 -04:00
Philippe Tillet
bfa85ef030 [GENERAL] Added compatibility with pytorch 1.2.0 and powerpc 2020-05-09 21:58:57 -04:00
jack-willturner
5fddc2062e [DOCS] Transposition fix 2020-05-07 14:02:18 +01:00
jack-willturner
91bfb0d182 Merge https://github.com/ptillet/triton 2020-05-07 13:29:44 +01:00
jack-willturner
be02315168 [DOCS] Matrix copy and transpose 2020-05-05 14:30:49 +01:00
Philippe Tillet
2dd3fcb270 [CORE] Fixed bug for Multi-GPU 2020-05-04 18:36:44 -04:00
jack-willturner
f5d47536c5 [DOCS] Matmul and vecadd working examples 2020-05-04 16:25:17 +01:00
Philippe Tillet
b9e36331be [PACKAGING] sdist now generates working .tar.gz file 2020-05-04 10:43:38 -04:00
Philippe Tillet
c3d4a8b7ae [PACKAGING] Fixed import error 2020-05-04 10:14:34 -04:00
Philippe Tillet
0685fcb804 [PACKAGING] Added MANIFEST.in and some symlinks for better packaging 2020-05-04 10:09:30 -04:00
Philippe Tillet
1e7fee573b [PACKAGING] Fixed typo in setup.py 2020-05-04 09:09:54 -04:00
Philippe Tillet
dc2ded4165 [PACKAGING] Added some more files for packaging 2020-05-04 08:58:58 -04:00
Philippe Tillet
4c2c244b91 [PYTHON][SRC][BINDING] Improved code portability across compilers 2020-04-17 13:51:47 -04:00
Philippe Tillet
eaabfb1d8e [PYTHON][EXAMPLES][EINSUM] Updated configs for matmul 2020-04-10 12:42:48 -04:00
Philippe Tillet
28f845eab1 [PYTHON][EXAMPLES][EINSUM] Added stride in CONV2D example 2020-04-10 00:14:31 -04:00
Philippe Tillet
8e276484ea [PYTHON][EXAMPLES][EINSUM] Added group-convolution test/benchmark 2020-04-09 23:37:39 -04:00