SIGN IN SIGN UP

Fix dtype mismatch in grouped_mm_fallback for LoRA training on Mamba+… (#47933)

* Fix dtype mismatch in grouped_mm_fallback for LoRA training on Mamba+MoE models

Ensure tensor dtype consistency in matrix multiplications.

* Apply ruff formatting

* Apply suggestions from code review

Co-authored-by: Ilyas Moutawwakil <57442720+IlyasMoutawwakil@users.noreply.github.com>

---------

Co-authored-by: Ilyas Moutawwakil <57442720+IlyasMoutawwakil@users.noreply.github.com>
A
Aakriti Adhikari committed
4ad919951a80657e9251f2a36c3fa245b5847a5f
Parent: a353632
Committed by GitHub <noreply@github.com> on 8/25/2026, 6:13:23 AM