InfiniTensor/src
zhengly123 1152adc94a
Add: python API for timing ConvTranspose (#46)
* Add: python interfaced for timing operators

* Fix: CUDA Runtime run

Co-authored-by: Liyan Zheng <liyan-zheng@outlook.com>
2022-10-07 16:03:11 +08:00
..
bang Add bangc runtime and element-wise kernels 2022-09-22 16:57:39 +08:00
core Add: python API for timing ConvTranspose (#46) 2022-10-07 16:03:11 +08:00
cuda Add: python API for timing ConvTranspose (#46) 2022-10-07 16:03:11 +08:00
ffi Add: python API for timing ConvTranspose (#46) 2022-10-07 16:03:11 +08:00
kernels Add: python API for timing ConvTranspose (#46) 2022-10-07 16:03:11 +08:00
nnet Fix CMake USE_CUDA (#36) 2022-09-21 12:28:00 +08:00
operators ADD:extend operator and cuda kernel. (#40) 2022-09-29 14:52:50 +08:00
utils Tensor serialization (#25) 2022-09-13 11:27:41 +08:00