InfiniTensor/include
wanghailu0717 0a5d273130 Add: print derivation steps for conv2gemm 2023-11-10 23:16:44 +08:00
..
bang fix bang runtime bug after merging distributed branch (#137) 2023-09-19 14:10:39 +08:00
core [Kunlun & CUDA & BANG] add depth2space operator (#178) 2023-11-10 17:58:26 +08:00
cuda Cuda softmax (#129) 2023-11-06 08:56:23 +08:00
ffi Add TVM codegen for MemboundOp (#35) 2022-09-22 18:06:45 +08:00
intelcpu Cpu backend2 (#77) 2023-04-17 12:15:23 +08:00
kunlun Xpu (#82) 2023-10-16 10:57:08 +08:00
nnet Add: print derivation steps for conv2gemm 2023-11-10 23:16:44 +08:00
operators [Kunlun & CUDA & BANG] add depth2space operator (#178) 2023-11-10 17:58:26 +08:00
utils tensor parallel for transformer (#125) 2023-09-14 14:19:45 +08:00
test.h Add python interface for CUDA operator evaluation (#42) 2022-09-27 10:41:12 +08:00