From eca84bdcb577bcb14d4a0b70c6f5ac0e4577b776 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 23 May 2026 08:44:37 +0000 Subject: [PATCH] fix: remove bad import in NVFP4 diag --- tests/unit/test_nvfp4_primitives.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/unit/test_nvfp4_primitives.py b/tests/unit/test_nvfp4_primitives.py index f4fa4aea..f0c3d2da 100644 --- a/tests/unit/test_nvfp4_primitives.py +++ b/tests/unit/test_nvfp4_primitives.py @@ -105,9 +105,10 @@ def test_nvfp4_primitives(): else: print(f" ❌ Scale factors are {x_sf.dtype} — unexpected!") - # Check what the gemm_runner actually passes to the kernel - from dsv4.ops.gemm_runner import Nvfp4GemmRunner - print(f" Nvfp4GemmRunner exists: {Nvfp4GemmRunner is not None}") + # Check the gemm_runner class name + import dsv4.ops.gemm_runner as gr + runner_classes = [name for name in dir(gr) if 'unner' in name or 'Gemm' in name] + print(f" gemm_runner classes: {runner_classes}") print() print("=" * 60)