InfiniTensor/test/kernels
xiaonans b0876a13ce
Merge branch 'master' into rope_and_silu
2024-02-04 10:57:36 +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 Xpu task4 support: add softmax (#172) 2023-10-30 16:01:05 +08:00
nativecpu feat: add reshape/identity/squeeze/flatten/unsqueeze op cpu kernel (#213) 2024-01-30 10:29:59 +08:00