InfiniTensor/test/kernels
xiaonans 965df4e294
[feature] add fused attention_kvcache operator support (#179)
* [feature] add fused attention_kvcache operator support

* add test to attention_kvcache op

* Add space line at EOF

---------

Co-authored-by: Haojie Wang <haojie0429@gmail.com>
2023-11-14 23:44:22 +08:00
..
bang Pooling ceil mode (#155) 2023-10-09 20:51:39 +08:00
cuda [feature] add fused attention_kvcache operator support (#179) 2023-11-14 23:44:22 +08:00
intelcpu memory_allocator (#103) 2023-08-13 13:39:35 +08:00
kunlun Xpu task4 support: add softmax (#172) 2023-10-30 16:01:05 +08:00
nativecpu add transpose, concat and split for native cpu (#158) 2023-10-12 10:14:28 +08:00