Philippe Tillet
|
d9945692a9
|
[dnn] better specification of recompilation key
|
2019-08-02 17:42:48 -07:00 |
|
Philippe Tillet
|
ead368d1ed
|
[general] a bunch of fixes in anticipation of proper triton vs cudnn
benchmarks
* DNN: Added partial auto-tuning mode and skeleton for heuristics
* Examples: Moduralized benchmarking and now evaluating ResNet-18 shapes
|
2019-07-21 20:17:56 -07:00 |
|
Philippe Tillet
|
b1d81a5802
|
more work on heuristics
|
2019-07-21 18:11:54 -07:00 |
|
Philippe Tillet
|
f0d8306437
|
[codegen/alignment_info] better handling of constants
|
2019-07-18 16:12:06 -07:00 |
|
Philippe Tillet
|
07c964919c
|
[dnn/shift] now strictly only shifting the interior
|
2019-07-16 20:18:48 -07:00 |
|
Philippe Tillet
|
28959fe165
|
[runtime/jit] made auto-tuning silent
|
2019-07-16 14:41:38 -07:00 |
|
Philippe Tillet
|
3e7a3ed67a
|
[dnn/shift]: added support for fp16
|
2019-07-13 21:05:34 -07:00 |
|
Philippe Tillet
|
f36a646ffc
|
[dnn/shift-conv] added and tested NCHW layout
|
2019-07-11 21:00:33 -07:00 |
|
Philippe Tillet
|
4ca83f1935
|
ugh bug in shift-conv striding
|
2019-07-10 17:00:22 -07:00 |
|
Philippe Tillet
|
f665c742f9
|
testing a simple shiftnet
|
2019-07-10 13:33:08 -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
|
b7986baffa
|
[dnn]: Now implementing all existing DNN routines using common base template and auto-tuner
|
2019-07-09 19:52:55 -07:00 |
|
Philippe Tillet
|
88675fa01a
|
[dnn] added base template class for mutualized auto-tuning
|
2019-07-09 16:09:34 -07:00 |
|
Philippe Tillet
|
f9db0449b7
|
[dnn] Adding batchnorm
|
2019-07-08 18:44:37 -07:00 |
|