Split Kernel Optimization for `group_index_select_or_add_2d_kernel`#142

Open

aryaman-gupta wants to merge 27 commits intomain_12162025_upstreamfrom

aryaman/group-index-optimizations

Commits on Dec 12, 2025

adds optimized path for small dimension sizes to group_index_select_or_add_2d_kernel
aryaman-gupta
committed

Commits on Dec 16, 2025

group_index_select_or_add_2d_kernel: splits into two kernels depending on embedding dim size
aryaman-gupta
committed
sparse_group_index.cu: edits some comments
aryaman-gupta
committed
adds USE_ROCM guards to subwarp optimizations for group_index_select_or_add_2d_kernel
aryaman-gupta
committed

Commits on Dec 18, 2025

Commits on Jan 7, 2026

sparse_group_index: corrects macro invoking group_index_select_or_add_2d_kernel_small kernel
aryaman-gupta
committed

Commits on Jan 12, 2026

Commits on Jan 30, 2026