InfiniTensor/pyinfinitensor
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
..
docs feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00
src/pyinfinitensor [feature] add fused attention_kvcache operator support (#179) 2023-11-14 23:44:22 +08:00
tests Xpu (#82) 2023-10-16 10:57:08 +08:00
pyproject.toml Change onnx-simplifier to onnxsim to resolve build issue on xpu (#164) 2023-10-21 02:58:32 +08:00