Charlie Fu 9cc373f390 [Kernel][Amd] Add fp8 kv cache support for rocm custom paged attention (#8577)

2024-09-19 17:37:57 +00:00

..

attention.cu

[Kernel][Amd] Add fp8 kv cache support for rocm custom paged attention (#8577)

2024-09-19 17:37:57 +00:00

ops.h

[Kernel][Amd] Add fp8 kv cache support for rocm custom paged attention (#8577)

2024-09-19 17:37:57 +00:00

torch_bindings.cpp

[Kernel][Amd] Add fp8 kv cache support for rocm custom paged attention (#8577)

2024-09-19 17:37:57 +00:00