[Dispatch] Don't sink collapse_shape through k2 dims#19379

Merged

IanWood1 merged 1 commit intoiree-org:mainfrom IanWood1:fix_attn_unit_dim_err

Dec 6, 2024

+78-2

Commits on Dec 5, 2024

[Dispatch] don't bubble reshapes through k2 dims
IanWood1
committed