pytorch/torch/testing
Simon Layton fb545fb068 Add MXFP4 grouped gemm support via. FBGEMM kernels (#166530)
Summary:

* Extend `_scaled_grouped_mm_v2` to include MXFP4 support
* Add testing to existing grouped routines

Test Plan:

```
pytest -svv -k "mxfp4 and group" 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/166530
Approved by: https://github.com/drisspg
2025-10-30 16:46:11 +00:00
..
_internal Add MXFP4 grouped gemm support via. FBGEMM kernels (#166530) 2025-10-30 16:46:11 +00:00
__init__.py Fix syntax for pyrefly errors (#166496) 2025-10-29 20:00:25 +00:00
_comparison.py Fix syntax for pyrefly errors (#166496) 2025-10-29 20:00:25 +00:00
_creation.py [BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552) 2025-08-07 00:09:56 +00:00
_utils.py