Files
pytorch/test
Simon Layton 7c6c5d04fe Add scaled_grouped_mm_v2 and python API (#165154)
Summary:

* Add `torch._scaled_grouped_mm_v2` with more functionality and
  extensibility for future formats
* Add `torch.nn.functional.scaled_grouped_mm` as public entrypoint
* Test both original and v2 functionality

Test Plan:

```
pytest -svv -k grouped test/test_scaled_matmul_cuda.py
```

Reviewers:

Subscribers:

Tasks:

Tags:
Signed-off-by: Simon Layton <simonlayton@meta.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165154
Approved by: https://github.com/drisspg, https://github.com/danielvegamyhre
2025-10-15 17:47:23 +00:00
..
2025-10-15 03:18:57 +00:00
2025-10-13 01:48:55 +00:00
2025-10-13 01:48:55 +00:00
2025-10-13 01:48:55 +00:00
2025-10-13 11:47:32 +00:00
2025-10-13 01:48:55 +00:00
2025-10-13 01:48:55 +00:00
2025-10-08 21:41:42 +00:00