InfiniTensor/include
whjthu 4e481e72a5 change cmake option for kunlunxin xpu from 'xpu' to 'kunlun'; fix bug after merging distributed infrastructure 2023-09-26 14:10:22 +08:00
..
bang Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
core Merge branch 'master' into xpu 2023-09-06 08:55:55 +08:00
cuda impl distributed launch with NCCL (#106) 2023-09-05 09:47:35 +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
nnet Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
operators impl distributed launch with NCCL (#106) 2023-09-05 09:47:35 +08:00
utils Add cuda transpose kernel (#115) 2023-08-22 14:22:15 +08:00
xpu change cmake option for kunlunxin xpu from 'xpu' to 'kunlun'; fix bug after merging distributed infrastructure 2023-09-26 14:10:22 +08:00
test.h Add python interface for CUDA operator evaluation (#42) 2022-09-27 10:41:12 +08:00