Update fp4fp6_helpers.py - #810
Open
devalshahamd wants to merge 4 commits into
Open
Conversation
devalshahamd
force-pushed
the
fix/fp4_avail_check
branch
from
July 14, 2026 23:08
efb0f66 to
3200e27
Compare
gabeweisz
reviewed
Jul 15, 2026
| Covers the Triton ``tl.dot_scaled`` MXFP4/MXFP6 path as well as aiter's | ||
| gfx950 CK ``gemm_a4w4`` MXFP4 path | ||
| """ | ||
| if triton_available() and (_MXFP4_SUPPORTED or bool(_MXFP6_DTYPE)): |
Collaborator
There was a problem hiding this comment.
Probably not worth changing but this could also be:
return (triton_available() and (_MXFP4_SUPPORTED or bool(_MXFP6_DTYPE))) or _aiter_mxfp4_ready()
Either way is fine if you don't want to make another update
Contributor
Author
There was a problem hiding this comment.
@gabeweisz Thanks fir the suggestion. I made the edit. Could you please reapprove?
gabeweisz
approved these changes
Jul 15, 2026
devalshahamd
enabled auto-merge (squash)
July 29, 2026 19:43
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
This pull request introduces a new utility function to check for the availability of block-scaled MX GEMM paths, improving the usability and clarity of hardware support checks in the codebase.
Hardware support detection:
mx_available()function infp4fp6_helpers.pyto determine if any block-scaled MX GEMM path (Tritontl.dot_scaledfor MXFP4/MXFP6 or aiter's CKgemm_a4w4for MXFP4) is usable.