From c652177970fe369e47ad9628e5bc8b3893e259cf Mon Sep 17 00:00:00 2001 From: biondizzle Date: Mon, 1 Jun 2026 04:35:55 +0000 Subject: [PATCH] test: fix gemm 1-group test --- test_gemm_1group.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test_gemm_1group.py b/test_gemm_1group.py index e75698be..1e3b7147 100644 --- a/test_gemm_1group.py +++ b/test_gemm_1group.py @@ -1,7 +1,7 @@ #!/usr/bin/env python3 """Test: run_nvfp4_grouped_gemm with num_groups=1 on different GPUs.""" import torch -from dsv4.ops.gemm_runner import run_nvfp4_grouped_gemm, warmup_nvfp4_compilation +from dsv4.ops.gemm_runner import run_nvfp4_grouped_gemm from dsv4.ops.quantize import quantize_nvfp4_gpu from dsv4.ops.layouts import make_b_k_major, assemble_scales_3d_side