Simon Layton 066f818eea Refactor and unify v1/v2 _scaled_mm codes (#165436) ...

Summary:

* Refactor out some core routines (scaled_gemm, auto-tuned scaled_gemm)
* Unify v1/v2 dispatch calls where possible
* Simplify call pattern w.r.t. CUDA/ROCM for easier readability.

Test Plan:

```
pytest -svv 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/165436
Approved by: https://github.com/drisspg

2025-10-15 19:07:05 +00:00

..

conda

PyTorch -> C++17 (#98209) (#100557)

2023-05-19 00:49:08 +00:00

src

Refactor and unify v1/v2 _scaled_mm codes (#165436)

2025-10-15 19:07:05 +00:00

tools

Adds Issue#153109 as a test for CUDAPluggableAllocator (#163575)

2025-10-01 09:07:48 +00:00

CMakeLists.txt

Revert "Use official CUDAToolkit module in CMake (#154595)"

2025-06-23 21:15:31 +00:00