Commit Graph

1262 Commits

Author SHA1 Message Date
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
Philippe Tillet
3848d25450 [DOCS] Added pip command in README.md 2020-05-07 10:48:41 -04:00
Philippe Tillet
1a20099582 [DOCS] Now specifying pip command in installation.rst 2020-05-07 10:45:58 -04:00
Phillippe Tillet
fc3747d59c Merge pull request #38 from jack-willturner/master
Add working examples to tutorials and python examples folder
2020-05-07 10:24:51 -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 v0.1 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
57a0b0a132 [CODEGEN] Fixed bug for phi nodes with constant incoming value 2020-05-02 17:30:15 -04:00
Philippe Tillet
a9efb27fde [CODEGEN][ANALYSIS] bugfix in alignment analysis 2020-05-01 17:38:23 -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
Philippe Tillet
840af73c8c [PYTHON][EINSUM] re-established auto-tuning 2020-04-09 11:01:57 -04:00
Philippe Tillet
25055cbba7 [TESTS] Simplified testing of half-precision transposes 2020-04-09 01:10:11 -04:00
Philippe Tillet
4114351c7c [PYTHON][KERNEL] Added thread-safety when caching custom torch op 2020-04-07 20:21:50 -04:00
Philippe Tillet
fa5e4af93e [CORE][RUNTIME] Better error message on internal compilation error 2020-04-07 14:01:21 -04:00
Philippe Tillet
0b35c5ff3f [PYTHON][KERNEL] Better handling of case where cache directory already
exists
2020-04-07 13:18:11 -04:00
Philippe Tillet
e04efc1c85 [GENERAL] Error messages now no longer make terminal color green 2020-04-03 23:25:29 -04:00
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
Phillippe Tillet
f68ed147fa Merge pull request #35 from jack-willturner/iss33
[PYTHON] Add empty string to llvm-config versions in setup.py
2020-03-13 15:06:32 -04:00
Jack Turner
53b9e4790e [PYTHON] Add empty string to llvm-config versions in setup.py 2020-03-13 18:03:25 +00:00
Philippe Tillet
699711724a [DOCS] Fixed typo: triton.function -> torch.autograd.Function 2020-03-13 11:44:55 -04:00
Philippe Tillet
7ee3380faf [DOCS] Made documentation consistent with the new kernel API 2020-03-10 13:25:57 -04:00
Philippe Tillet
2dcf381bdc [CORE][TOOLS][BENCH] Disabled normalization for auto-tuning 2020-03-06 18:41:36 -05:00
Philippe Tillet
a27b96cad1 [CORE][DRIVER] Now only using PTX6.4 if CUDA10.1+ is detected 2020-03-05 21:29:22 -05:00
Philippe Tillet
20c0246d88 [CMAKE] target_link_directories -> link_directories 2020-03-05 20:16:46 -05:00
Philippe Tillet
b7895c653f [PYTHON][EXAMPLES] Removed BlockSparse examples; see
https://github.com/ptillet/torch-blocksparse.git
2020-03-05 13:32:42 -05:00
Philippe Tillet
1f1e4ee9ec [PYTHON] Merged blocksparse branch:
* Example for blocksparse matrix multiplication
* Simplified Triton kernel API
* Revived auto-tuning in einsum
2020-03-05 13:08:07 -05:00
Philippe Tillet
51bc244b2c [PYTHON][OPS] Fixed typo in einsum 2020-02-25 16:39:04 -05:00
Philippe Tillet
cc6892f9f4 [CMAKE] Fixed issue in LLVM link directory 2020-02-25 12:52:23 -08:00
Philippe Tillet
3ee342d056 [TRITON][NN][CONV] Renamed input -> x to not modify built-in functions 2020-02-25 10:56:39 -05:00
Philippe Tillet
1e76bb4145 [PYTHON][NN][CONV] Fixed typo in dx computation 2020-02-24 22:45:49 -05:00
Philippe Tillet
cc094936d0 [PYTHON][SETUP] Removed obsolete debug print() 2020-02-24 21:35:45 -05:00
Philippe Tillet
d0ca849be2 [DOCUMENTATION] Updated installation instructions in README.md 2020-02-24 18:07:25 -05:00
Philippe Tillet
a59c24ab67 [CORE] Added missing sha1.hpp file 2020-02-24 18:02:18 -05:00
Philippe Tillet
8c35bd775f [PYTHON] Added missing files for nn submodule 2020-02-24 17:58:24 -05:00
Philippe Tillet
67c633aa2d [PYTHON] Better packaging 2020-02-24 17:46:20 -05:00
Philippe Tillet
f2daff85d2 [GENERAL] Improved caching mechanism:
* Now computing hash in libtriton
* Now only compiling a single pytorch hook per function signature
2020-02-24 16:36:50 -05:00
Philippe Tillet
29c38b38e5 [PYTHON][OPS][EINSUM] Now throwing error for automatic differentiation
of extended einsum
2020-02-20 17:29:01 -05:00
Philippe Tillet
9693fe1441 [PYTHON][OP][EINSUM] simplified API 2020-02-19 23:42:22 -05:00
Philippe Tillet
c30c4861be [PYTHON][OPS][EINSUM] Added support for inner tensor strides 2020-02-19 11:50:17 -05:00