Philippe Tillet
|
d9945692a9
|
[dnn] better specification of recompilation key
|
2019-08-02 17:42:48 -07:00 |
|
Philippe Tillet
|
3b89bc8463
|
[examples/python/pytorch] added skeleton of wrapper for shift-conv and batch-norm
|
2019-07-09 21:54:37 -07:00 |
|
Philippe Tillet
|
63b249c1d6
|
[examples/python/pytorch] added batchnorm cpp extension
|
2019-07-09 20:59:04 -07:00 |
|
Philippe Tillet
|
f9db0449b7
|
[dnn] Adding batchnorm
|
2019-07-08 18:44:37 -07:00 |
|
Philippe Tillet
|
a9d078c06f
|
[triton/dnn/conv] merged optimizations branch
- Added forward/backward support for strided convolution
- Added support for bias
- Added support for reduction splitting
|
2019-05-28 14:04:53 -04:00 |
|
Philippe Tillet
|
e8f23bcade
|
[dnn/conv] Added bias and forward stride
|
2019-05-22 13:27:08 -04:00 |
|
Philippe Tillet
|
f33a1f3fe3
|
[examples/pytorch] Fixed issues in backward pass of conv
|
2019-05-19 01:31:08 -04:00 |
|
Philippe Tillet
|
b2b55c52c9
|
[triton/python/conv]: Added cache for compiled kernels
|
2019-05-18 11:51:49 -04:00 |
|
Philippe Tillet
|
600aef72d5
|
[conv/dnn] now created a separate .h and .cpp file
|
2019-05-17 12:29:11 -04:00 |
|
Philippe Tillet
|
5941501f70
|
[dnn] added Triton-C derivative computations in conv
|
2019-05-13 18:04:11 -04:00 |
|
Philippe Tillet
|
fd91368f98
|
[general] creation of dnn module for gemm/conv triton routines
|
2019-05-06 17:47:06 -04:00 |
|
Philippe Tillet
|
f80441017c
|
[codegen] added leading dimension padding for transposition in shared
memory
|
2019-05-06 11:53:35 -04:00 |
|
Philippe Tillet
|
30833c18f1
|
[codegen/tune] bugfix in heuristics for nano-tile sizes
|
2019-05-04 01:32:34 -04:00 |
|
Philippe Tillet
|
0d694445e6
|
[examples] added skeleton for pytorch wrapper
|
2019-05-03 14:30:06 -04:00 |
|