Skip to content

Add block size table for ragged_paged_attention#8942

Merged
yaochengji merged 4 commits intomasterfrom
chengji/fix-ragged-attn
Apr 5, 2025
Merged

Add block size table for ragged_paged_attention#8942
yaochengji merged 4 commits intomasterfrom
chengji/fix-ragged-attn

Conversation

@yaochengji
Copy link
Copy Markdown
Collaborator

No description provided.

@yaochengji yaochengji requested a review from vanbasten23 April 5, 2025 01:17
Comment thread torch_xla/experimental/custom_kernel.py Outdated
Comment thread torch_xla/experimental/tuned_block_sizes.py
Comment thread torch_xla/experimental/tuned_block_sizes.py
Copy link
Copy Markdown
Collaborator

@vanbasten23 vanbasten23 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks Chengji!

@yaochengji yaochengji enabled auto-merge (squash) April 5, 2025 05:56
@yaochengji yaochengji merged commit cbff76f into master Apr 5, 2025
21 of 24 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants