pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Simon Layton 6a7f5c0d21 Add scaled_mm python API, test (#164142 )

Summary:

* Add `torch.nn.functional.scaled_mm` as an abstraction around the C++
  methods
* Wraps `torch._scaled_mm_v2` API by default, but user can force use of
  the older `torch._scaled_mm` interface.
* Scaled MM tests now run on the new API

Test Plan:

`pytest test/test_scaled_matmul_cuda.py`

Reviewers:

Subscribers:

Tasks:

Tags:
Signed-off-by: Simon Layton <simonlaytonmeta.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164142
Approved by: https://github.com/drisspg
ghstack dependencies: #164141

2025-10-09 12:43:18 +00:00

cpp

Fix cpp build (#162774 )

2025-09-25 18:21:45 +00:00

source

Add scaled_mm python API, test (#164142 )

2025-10-09 12:43:18 +00:00

.gitignore

.gitignore for the docs folder

2019-10-08 12:18:30 -07:00

libtorch.rst

Add ROCm documentation to libtorch (C++) reST. (#136378 )