pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00

History

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
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

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