[Tests] Update test_fp8_quant.py (#337)

dsikka · web-flow · commit bbe5491bad8f · 2025-06-02T12:09:12.000-04:00
- Fix g_idx init
diff --git a/tests/test_compressors/quantized_compressors/test_fp8_quant.py b/tests/test_compressors/quantized_compressors/test_fp8_quant.py
@@ -79,7 +79,7 @@ def test_quant_format(strategy, group_size, sc, zp):
         "dummy.weight_zero_point": torch.tensor(zp, dtype=torch.float32),
     }
     if group_size is not None:
-        dense_state_dict["dummy.weight_g_idx"] = make_dummy_g_idx(512, group_size)
+        dense_state_dict["dummy.weight_g_idx"] = make_dummy_g_idx(1024, group_size)
 
     quant_config = get_dummy_quant_config(strategy=strategy, group_size=group_size)
 

Original file line number	Diff line number	Diff line change
`@@ -79,7 +79,7 @@ def test_quant_format(strategy, group_size, sc, zp):`
`79`	`79`	`"dummy.weight_zero_point": torch.tensor(zp, dtype=torch.float32),`
`80`	`80`	`}`
`81`	`81`	`if group_size is not None:`
`82`		`- dense_state_dict["dummy.weight_g_idx"] = make_dummy_g_idx(512, group_size)`
	`82`	`+ dense_state_dict["dummy.weight_g_idx"] = make_dummy_g_idx(1024, group_size)`
`83`	`83`
`84`	`84`	`quant_config = get_dummy_quant_config(strategy=strategy, group_size=group_size)`
`85`	`85`