InfiniTensor/src
zhangyue 5559536470
add kunlun squeeze kernel (#229)
Co-authored-by: Haojie Wang <haojie0429@gmail.com>
2024-04-28 11:28:28 +08:00
..
bang Modify kernel registration & support fp16 (#205) 2024-01-15 11:02:13 +08:00
core Accelerate llama (#219) 2024-04-01 08:46:05 +08:00
cuda [feature] add cudagraph support (#215) 2024-02-21 14:00:25 +08:00
ffi Accelerate llama (#219) 2024-04-01 08:46:05 +08:00
intelcpu Cpu backend2 (#77) 2023-04-17 12:15:23 +08:00
kernels add kunlun squeeze kernel (#229) 2024-04-28 11:28:28 +08:00
kunlun XCCL support (#171) 2024-02-29 11:48:35 +08:00
nnet test: 支持编译 einnet 单元测试,但不是所有测试都能通过 (#174) 2023-11-03 13:21:49 +08:00
operators Accelerate llama (#219) 2024-04-01 08:46:05 +08:00
utils XCCL support (#171) 2024-02-29 11:48:35 +08:00