InfiniTensor/src
panzezhong 0ce7e7651f Fix kernel arguments, add debug mode 2023-08-24 13:39:22 +08:00
..
bang fix: fix cuda conv_fp16 run fail (#105) 2023-08-10 15:22:18 +08:00
core add CUDNN impl for Min and Max (#118) 2023-08-22 16:19:29 +08:00
cuda Fix kernel arguments, add debug mode 2023-08-24 13:39:22 +08:00
ffi impl sqrt on CUDA (#109) 2023-08-18 12:17:47 +08:00
intelcpu Cpu backend2 (#77) 2023-04-17 12:15:23 +08:00
kernels Fix kernel arguments, add debug mode 2023-08-24 13:39:22 +08:00
nnet NNET supports TVM backend and kernels (#78) 2023-04-18 00:26:36 +08:00
operators impl sqrt on CUDA (#109) 2023-08-18 12:17:47 +08:00
utils support mixed dtype (#102) 2023-08-16 21:49:43 +08:00