test: fix gemm 1-group test

2026-06-01 04:35:55 +00:00
parent 793f062bbc
commit c652177970
1 changed files with 1 additions and 1 deletions
--- 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