InfiniTensor/include
xgqdut2016 1ed4b36db2 add bangSoftmax , compare cnnl and bang C 2024-02-28 03:14:58 +00:00
..
bang add bangSoftmax , compare cnnl and bang C 2024-02-28 03:14:58 +00:00
core add bangSoftmax , compare cnnl and bang C 2024-02-28 03:14:58 +00:00
cuda [feature] add cudagraph support (#215) 2024-02-21 14:00:25 +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 test: 支持编译 einnet 单元测试,但不是所有测试都能通过 (#174) 2023-11-03 13:21:49 +08:00
operators add bangSoftmax , compare cnnl and bang C 2024-02-28 03:14:58 +00:00
utils Modify kernel registration & support fp16 (#205) 2024-01-15 11:02:13 +08:00
test.h Add python interface for CUDA operator evaluation (#42) 2022-09-27 10:41:12 +08:00