Add ragged paged attention (#8659)

pytorch · Feb 4, 2025 · 8480094 · 8480094
1 parent 1c53790
commit 8480094
Show file tree

Hide file tree

Showing 4 changed files with 1,359 additions and 1 deletion.
diff --git a/test/test_tpu_paged_attention_kernel.py → ...t_multi_queries_paged_attention_kernel.py b/test/test_tpu_paged_attention_kernel.py → ...t_multi_queries_paged_attention_kernel.py