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
|
28c250216c
|
[dnn/gemm] added some bounds checking
|
2019-07-19 21:32:55 -07:00 |
|
Philippe Tillet
|
5215fb0424
|
[codegen] some more optimizations
|
2019-07-19 20:29:03 -07:00 |
|
Philippe Tillet
|
f0d8306437
|
[codegen/alignment_info] better handling of constants
|
2019-07-18 16:12:06 -07:00 |
|
Philippe Tillet
|
bfa39b8992
|
preparing the field for tensor cores transposes
|
2019-07-17 13:20:33 -07:00 |
|
Philippe Tillet
|
a55b098e88
|
[dnn/shift] now using constant divisions
|
2019-07-16 21:05:21 -07:00 |
|
Philippe Tillet
|
07c964919c
|
[dnn/shift] now strictly only shifting the interior
|
2019-07-16 20:18:48 -07:00 |
|
Philippe Tillet
|
164d85077f
|
more stuff
|
2019-07-16 15:03:53 -07:00 |
|
Philippe Tillet
|
28959fe165
|
[runtime/jit] made auto-tuning silent
|
2019-07-16 14:41:38 -07:00 |
|
Philippe Tillet
|
7d1797cd32
|
ugh
|
2019-07-16 12:59:27 -07:00 |
|
Philippe Tillet
|
f50d7a420a
|
[runtime/jit] fixed bug in multi-threaded auto-tuning
|
2019-07-15 21:16:50 -07:00 |
|
Philippe Tillet
|
aa8bcf6bde
|
[dnn/shift] added split-k for shift-conv
|
2019-07-15 21:03:58 -07:00 |
|
Philippe Tillet
|
434f65737f
|
[runtime] put jit::launch_info in another file
|
2019-07-15 12:35:53 -07:00 |
|
Philippe Tillet
|
3c128fc2e2
|
[jit/autotune] added support for multi-threaded auto-tuning
|
2019-07-14 22:31:30 -07:00 |
|
Philippe Tillet
|
3e7a3ed67a
|
[dnn/shift]: added support for fp16
|
2019-07-13 21:05:34 -07:00 |
|
Philippe Tillet
|
1d88f0a36b
|
stuff
|
2019-07-03 19:25:16 -07:00 |
|
Philippe Tillet
|
d945ce5e1b
|
Now showing valid parameter for NN
|
2019-06-25 19:18:43 -07:00 |
|
Philippe Tillet
|
06b5992509
|
[feature] added basic tensor core support
|
2019-06-11 10:24:49 -07:00 |
|
Philippe Tillet
|
781b6d377d
|
seleciton now segfault (expected
|
2019-06-06 20:34:56 -07:00 |
|
Philippe Tillet
|
6045209d5b
|
Now find correct tuning configuration
|
2019-06-06 20:13:26 -07:00 |
|
Philippe Tillet
|
383b5b2a2a
|
[triton/ast] renamed ast -> lang in namespace and file structure
|
2019-05-28 17:28:02 -04: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
|
f33a1f3fe3
|
[examples/pytorch] Fixed issues in backward pass of conv
|
2019-05-19 01:31:08 -04:00 |
|