Skip to content

Disable reduced precision reductions in FP16 and BF16 GEMMs#51

Merged
voltjia merged 1 commit intomasterfrom
disable-reduced-precision-reductions-in-fp16-and-bf16-gemms
Oct 13, 2025
Merged

Disable reduced precision reductions in FP16 and BF16 GEMMs#51
voltjia merged 1 commit intomasterfrom
disable-reduced-precision-reductions-in-fp16-and-bf16-gemms

Conversation

@voltjia
Copy link
Collaborator

@voltjia voltjia commented Oct 13, 2025

pytest output:

============================= test session starts ==============================
platform linux -- Python 3.12.3, pytest-8.3.5, pluggy-1.5.0
rootdir: /home/voltjia/ntops
configfile: pyproject.toml
plugins: jaxtyping-0.3.2, anyio-4.9.0, cov-6.0.0
collected 6 items

tests/test_addmm.py ..                                                   [ 33%]
tests/test_bmm.py ..                                                     [ 66%]
tests/test_mm.py ..                                                      [100%]

======================== 6 passed in 388.29s (0:06:28) =========================

@voltjia voltjia merged commit 75445b5 into master Oct 13, 2025
4 checks passed
@voltjia voltjia deleted the disable-reduced-precision-reductions-in-fp16-and-bf16-gemms branch October 13, 2025 08:09
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant