lightning-thunder icon indicating copy to clipboard operation
lightning-thunder copied to clipboard

Published 20 hours ago • verified publisher icon Lightning-AI

Add `torch.nn.functional.scaled_grouped_mm`

Open crcrpar opened this issue 4 weeks ago • 0 comments

What does this PR do?

As per title. This PR adds torch.nn.functional.scaled_grouped_mm support.

ref: https://docs.pytorch.org/docs/main/generated/torch.nn.functional.scaled_grouped_mm.html

crcrpar avatar Nov 03 '25 14:11 crcrpar