diff --git a/dsv4/ops/gemm_runner.py b/dsv4/ops/gemm_runner.py index e9ec40e9..a7d24b68 100644 --- a/dsv4/ops/gemm_runner.py +++ b/dsv4/ops/gemm_runner.py @@ -13,6 +13,7 @@ from dsv4.ops.quantize import ( quantize_weight_to_nvfp4, quantize_to_nvfp4, deinterleave_quantize_nvfp4_cuda, + SF_VEC_SIZE, ) from dsv4.ops.layouts import ( interleave_l1_weights,