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
|
e9c787ef05
|
[PYTHON][EINSUM] Added support for FP16
|
2019-10-28 14:07:17 -04:00 |
|
Philippe Tillet
|
0ec213547c
|
[PYTHON][KERNEL] Added benchmarking functionalities for kernels
|
2019-10-28 00:30:04 -04:00 |
|
Philippe Tillet
|
96bdae25d5
|
[python][example] now executing tensorflow and/or pytorch example
automatically
|
2019-09-05 21:35:23 -04:00 |
|
Philippe Tillet
|
9ab2880fba
|
[python][examples] cleaned up dot example
|
2019-09-05 12:54:35 -04:00 |
|
Philippe Tillet
|
2d6c8311e8
|
[python] upgraded pybind11 ; forcing torch tensors to be contiguous()
|
2019-09-05 12:30:51 -04:00 |
|
Philippe Tillet
|
44896ee777
|
[pytorch] clean-up of dynamic framework load
|
2019-09-05 02:16:27 -04:00 |
|
Philippe Tillet
|
65133cdf33
|
[python] basic support for pytorch seems to be working
|
2019-09-05 01:32:21 -04:00 |
|
Philippe Tillet
|
ed0f706005
|
[python] fixed various issues in pytorch supoport
|
2019-09-05 00:19:42 -04:00 |
|
Philippe Tillet
|
945b5d0de9
|
[python] modularized triton package
|
2019-09-04 21:55:47 -04:00 |
|
Philippe Tillet
|
f6e9c24fe8
|
[python] more progress towards tensorflow/pytorch unification
|
2019-09-04 19:45:50 -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
|
5db3a7adfe
|
[python][examples] some more cleaning of dot product example
|
2019-08-30 17:05:03 -07:00 |
|
Philippe Tillet
|
7e0af2118c
|
[codegen] worked around bug seemingly from nvptx/ptxas by simplifying multiplications by 1:
- Generated LLVM-IR looked correct
- Illegal addressing disappeared when running cuda-memcheck
- Illegal addressing disappeared when using nvptx-short-pointer
|
2019-08-30 16:45:14 -07:00 |
|
Philippe Tillet
|
141a823799
|
[python] refactoring in anticipation of pytorch support
|
2019-08-29 18:08:51 -07:00 |
|
Philippe Tillet
|
7cb73f66e2
|
testing some register gradient
|
2019-08-26 19:25:58 -07:00 |
|
Philippe Tillet
|
4075949f80
|
[python] basic tensorflow wrapper working
|
2019-08-26 16:53:49 -07:00 |
|
Philippe Tillet
|
321d268a4a
|
more progress
|
2019-08-25 21:26:09 -07:00 |
|
Philippe Tillet
|
81571246cf
|
[general] fixed some warnings
|
2019-08-18 14:08:57 -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
|
c7cb5f82ad
|
[general] removed LLVM #include's in all Triton headers
|
2019-08-16 15:56:58 -07:00 |
|
Philippe Tillet
|
4de22df930
|
[python] added skeleton for python interface
|
2019-08-15 20:50:10 -07:00 |
|