From 3200e27a74b49b0a2089de711004786a7846ef99 Mon Sep 17 00:00:00 2001 From: devalshahamd Date: Tue, 14 Jul 2026 16:02:36 -0700 Subject: [PATCH 1/3] Update fp4fp6_helpers.py --- TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py b/TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py index 0aa420b6..f605667f 100644 --- a/TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py +++ b/TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py @@ -33,6 +33,16 @@ MX_BLOCK = 32 +def mx_available() -> bool: + """True if any block-scaled MX GEMM path is usable. + 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)): + return True + return _aiter_mxfp4_ready() + + def triton_available() -> bool: return triton is not None and tl is not None From 348ff07803e1cfa27cdb8d714d3d37719f84f6ba Mon Sep 17 00:00:00 2001 From: devalshahamd Date: Wed, 29 Jul 2026 12:38:39 -0700 Subject: [PATCH 2/3] Update fp4fp6_helpers.py --- TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py b/TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py index f605667f..84091f7d 100644 --- a/TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py +++ b/TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py @@ -38,9 +38,7 @@ def mx_available() -> bool: 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)): - return True - return _aiter_mxfp4_ready() + return (triton_available() and (_MXFP4_SUPPORTED or bool(_MXFP6_DTYPE))) or _aiter_mxfp4_ready() def triton_available() -> bool: From 008c495098887be3f1c99557080331e99a2a45ea Mon Sep 17 00:00:00 2001 From: Deval Shah Date: Wed, 29 Jul 2026 14:41:55 -0500 Subject: [PATCH 3/3] reformatting --- TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py b/TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py index 84091f7d..cf950626 100644 --- a/TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py +++ b/TraceLens/PerfModel/benchmarking/fp4fp6_helpers.py @@ -38,7 +38,9 @@ def mx_available() -> bool: Covers the Triton ``tl.dot_scaled`` MXFP4/MXFP6 path as well as aiter's gfx950 CK ``gemm_a4w4`` MXFP4 path """ - return (triton_available() and (_MXFP4_SUPPORTED or bool(_MXFP6_DTYPE))) or _aiter_mxfp4_ready() + return ( + triton_available() and (_MXFP4_SUPPORTED or bool(_MXFP6_DTYPE)) + ) or _aiter_mxfp4_ready() def triton_available() -> bool: