Commit 7a3c051

authored

[ragged-paged-attn] Combine k_pages and v_pages on num_kv_head (#8892)

1 parent 96ad8f5 commit 7a3c051Copy full SHA for 7a3c051

3 files changed

+389

-333

lines changed

test
- test_pallas.py
torch_xla/experimental
- custom_kernel.py
- pallas_kernels
  - ragged_paged_attention_v2.py

3 files changed

+389

-333

lines changed

Comments

(0)