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
6a4d42c1b8
[PYTHON][CORE] Deprecating Tensorflow support
2020-02-10 04:20:33 -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
f4bbbbe5e4
[PYTHON][OPS] Bugfix in conv fprop
2019-11-01 00:43:02 -04:00
Philippe Tillet
739a8d9061
some work on conv
2019-10-31 18:08:27 -04:00
Philippe Tillet
e0fe8d9058
[PYTHON][TENSORFLOW] More work
2019-10-30 18:39:58 -04:00
Philippe Tillet
9b0f1a0807
more stuff
2019-10-30 13:44:31 -04:00
Philippe Tillet
bf3dc63858
[PYTHON] Removed dead code for alloc_empty and register_scalar
2019-10-30 10:37:30 -04:00
Philippe Tillet
2b9355c9e4
[PYTHON][TENSORFLOW] Got rid of alloc_empty entirely; now doing
...
generating allocation code inside the tensorflow op
2019-10-30 01:38:30 -04:00
Philippe Tillet
d65a94c768
[PYTHON][OPS] Added batch normalization op
2019-10-29 17:29:11 -04:00
Philippe Tillet
448f4433d9
[PYTHON][KERNEL] Enforcing shapes to be known at compile-time for
...
TensorFlow Graph Execution
2019-10-29 00:48:53 -04:00
Philippe Tillet
0ec213547c
[PYTHON][KERNEL] Added benchmarking functionalities for kernels
2019-10-28 00:30:04 -04:00
Philippe Tillet
e827d4f467
[python] [bindings] removed obsolete #include
2019-10-20 20:37:37 -04:00
Philippe Tillet
1f8fd525b5
[python] fixed warnings for pybind11 and pytorch
2019-09-05 20:28:00 -04:00
Philippe Tillet
2d6c8311e8
[python] upgraded pybind11 ; forcing torch tensors to be contiguous()
2019-09-05 12:30:51 -04:00
Philippe Tillet
58544d0523
[python] renamed src/tensorflow.cc -> src/bindings.cc
2019-09-05 09:39:58 -04:00
Philippe Tillet
ed0f706005
[python] fixed various issues in pytorch supoport
2019-09-05 00:19:42 -04:00
Philippe Tillet
cdbc9d4ecd
[python] more generic gradient registration
2019-09-04 03:12:23 -04:00
Philippe Tillet
b747959a57
trying to work around tensorflow limitations
2019-09-04 01:54:43 -04:00
Philippe Tillet
2ccc915011
[python][examples] added template for blocksparse
2019-09-03 20:44:27 -04:00
Philippe Tillet
141a823799
[python] refactoring in anticipation of pytorch support
2019-08-29 18:08:51 -07:00
Philippe Tillet
d457482539
[codegen] fixed issue in double buffering pointer update
2019-08-28 17:50:45 -07:00
Philippe Tillet
7cb73f66e2
testing some register gradient
2019-08-26 19:25:58 -07:00
Philippe Tillet
9ece3eccc6
some cleaning
2019-08-26 17:28:24 -07:00
Philippe Tillet
4075949f80
[python] basic tensorflow wrapper working
2019-08-26 16:53:49 -07:00
Philippe Tillet
0e0399f866
more tests
2019-08-26 11:00:00 -07:00
Philippe Tillet
321d268a4a
more progress
2019-08-25 21:26:09 -07:00
Philippe Tillet
732156b942
[general] rename *.cpp -> *.cc
2019-08-23 19:06:39 -07:00
Philippe Tillet
bc11e31419
[lang] more progress on parser
2019-08-19 20:56:39 -07:00
Philippe Tillet
0970fe12dd
[general] cleaned tensorflow source code generation
2019-08-18 15:39:36 -07:00
Philippe Tillet
b4a9ed9663
[python] added basic tensorflow support
2019-08-17 18:18:26 -07:00
Philippe Tillet
078f0052fe
more cleaning
2019-08-17 16:12:17 -07:00
Philippe Tillet
11a6a92598
[python][tensorflow] basic op generation is working
2019-08-16 20:50:18 -07:00
Philippe Tillet
4de22df930
[python] added skeleton for python interface
2019-08-15 20:50:10 -07:00
