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

Published 20 hours ago • verified publisher icon Lightning-AI

Add `F.scaled_mm`

Open crcrpar opened this issue 1 month ago • 0 comments

What does this PR do?

As per title, this PR adds F.scaled_mm to thunder.torch and cover it with torchex impl.

Ref: https://docs.pytorch.org/docs/main/generated/torch.nn.functional.scaled_mm.html

crcrpar avatar Nov 06 '25 04:11 crcrpar