Add torch.backends.cuda.math_sdp.fp32_precision#2848

Merged

anatoliylitv merged 11 commits intorocm7.2_internal_testingfrom

anatoliylitv/rocm_7_2_internal_testing_Add-torch.backends.cuda.math_sdp.fp32_precision

Dec 5, 2025

Commits on Dec 2, 2025

Add torch.backends.cuda.math_sdp.fp32_precision

xinyazhang
authored and
anatoliylitv
committed
Make torch.backends.cuda.math_sdp.fp32_precision effective for math_sdp

xinyazhang
authored and
anatoliylitv
committed
torch/testing: add ctx mananger math_sdp_precision

xinyazhang
authored and
anatoliylitv
committed
test/test_transformers: decorate all tests that uses fp32 math as golden to use ieee rather than tf32

xinyazhang
authored and
anatoliylitv
committed
fix build error

xinyazhang
authored and
anatoliylitv
committed
test/test_transformers: sanity check of golden tensor

xinyazhang
authored and
anatoliylitv
committed
fix Fp32PrecisonGuard

xinyazhang
authored and
anatoliylitv
committed
more documentation

xinyazhang
authored and
anatoliylitv
committed
fix lint

xinyazhang
authored and
anatoliylitv
committed
Follow the practice of cuBLASModule and ignore MathSDPModule in docs generation

xinyazhang
authored and
anatoliylitv
committed

Commits on Dec 3, 2025

Follow the practice of public cuBLASModule to include to __all__ for MathSDPModule
anatoliylitv
committed