OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
7c6c5d04fe
pytorch/docs
History
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
..
cpp Fix cpp build (#162774) 2025-09-25 18:21:45 +00:00
source Add scaled_grouped_mm_v2 and python API (#165154) 2025-10-15 17:47:23 +00:00
.gitignore
libtorch.rst Add ROCm documentation to libtorch (C++) reST. (#136378) 2024-09-25 02:30:56 +00:00
make.bat
Makefile [ONNX] Filter out torchscript sentences (#158850) 2025-07-24 20:59:06 +00:00
README.md
requirements.txt Revert "Switch to standard pep517 sdist generation (#152098)" 2025-07-01 14:14:52 +00:00

README.md

Please see the Writing documentation section of CONTRIBUTING.md for details on both writing and building the docs.