fix bug

Signed-off-by: xin3he <[email protected]>
intel · Jun 21, 2024 · c7808e4 · c7808e4
1 parent 33bb948
commit c7808e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/3x/torch/quantization/weight_only/test_rtn.py b/test/3x/torch/quantization/weight_only/test_rtn.py
@@ -241,7 +241,7 @@ def test_double_quant_params(self, dtype, double_quant_bits, double_quant_group_
         out = model(self.example_inputs)[0]
         atol_true = (out - self.q_label).amax()
         # compare atol, this case is an ideal case.
-        if not (dtype, double_quant_bits, double_quant_group_size) == (256, 6, "nf4"):
+        if not (dtype, double_quant_bits, double_quant_group_size) == ("nf4", 6, 256):
             assert (
                 atol_false < atol_true
             ), "asym for double quant should have smaller atol because scales is bigger than zero, please double check."