InfiniTensor/test/kernels
OdinaryWord 6ba1a0648a add layernorm 2024-04-26 15:25:41 +08:00
..
ascend add layernorm 2024-04-26 15:25:41 +08:00
bang fix mlu some kernel registration & gather op (#210) 2024-02-01 15:02:02 +08:00
cuda add test for rotary embedding cuda kernel 2024-02-04 10:24:20 +08:00
intelcpu Modify kernel registration & support fp16 (#205) 2024-01-15 11:02:13 +08:00
kunlun XCCL support (#171) 2024-02-29 11:48:35 +08:00
nativecpu feat: add reshape/identity/squeeze/flatten/unsqueeze op cpu kernel (#213) 2024-01-30 10:29:59 +08:00