InfiniTensor/test
xiaonans 9a3c0f11f6 add test for rotary embedding cuda kernel 2024-02-04 10:24:20 +08:00
..
bang Bang cncl (#163) 2024-01-03 13:28:03 +08:00
core Add ReduceSum op and kernel (#160) 2023-11-24 09:29:58 +08:00
cuda impl distributed launch with NCCL (#106) 2023-09-05 09:47:35 +08:00
kernels add test for rotary embedding cuda kernel 2024-02-04 10:24:20 +08:00
nnet test: 支持编译 einnet 单元测试,但不是所有测试都能通过 (#174) 2023-11-03 13:21:49 +08:00
operators Modify kernel registration & support fp16 (#205) 2024-01-15 11:02:13 +08:00
script build: 实现格式化 git added c/c++ 源码的脚本 (#98) 2023-07-21 12:29:50 +08:00