Skip to content

Optimize group_index_select_or_add_2d_kernel on ROCm by adding a separate codepath for small embedding dimensions #5233

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
aryaman-gupta wants to merge 6 commits into
base: main
Choose a base branch
from
+96 −18

group_index_select_or_add_2d_kernel: when num_cols < UNROLL_FACTOR, d…

e0edc40
Select commit
Loading
Failed to load commit list.
Open

Optimize group_index_select_or_add_2d_kernel on ROCm by adding a separate codepath for small embedding dimensions #5233

group_index_select_or_add_2d_kernel: when num_cols < UNROLL_FACTOR, d…
e0edc40
Select commit
Loading
Failed to load commit list.
Meta CodeSync / Import Status succeeded Dec 18, 2025 in 4h 12m 34s

PR has been imported

@q10 has imported this pull request.

View more details on Meta CodeSync