Simon Layton d14cbb4476 Add NVFP4 two-level scaling to scaled_mm (#165774) ...

Summary:

* Add second-level scaling dispatch to scaled_mm, tying into optional `alpha` passing
* Add two-level tests

Test Plan:

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

2025-10-18 13:06:04 +00:00

..

conda

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

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

src

Add NVFP4 two-level scaling to scaled_mm (#165774)

2025-10-18 13:06:04 +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