Skip to content
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions test/3x/torch/quantization/weight_only/test_gptq.py
Original file line number Diff line number Diff line change
Expand Up @@ -223,9 +223,9 @@ def test_true_sequential(self):
out = model(self.example_inputs)[0]
atol_true = (out - self.label).amax()
# compare atol, this case is an ideal case.
assert (atol_false < atol_true
), "true_sequential=True doesn't help accuracy, maybe is reasonable, please double check."

assert (
atol_false < atol_true
), "true_sequential=True doesn't help accuracy, maybe is reasonable, please double check."

@pytest.mark.parametrize("dtype", ["nf4", "int4"])
@pytest.mark.parametrize("double_quant_bits", [6])
Expand Down