Skip to content

Commit 62ee666

Browse files
TST Mark AutoAWQ as xfail for now (#2529)
The AutoAWQ multi GPU test is currently failing on CI. This is most likely an issue of AutoAWQ with PyTorch 2.7. The issue has been reported but there is no reaction so far. Thus let's skip the test for the time being. Since the PR marks the test as strictly x-failing, we will know when there is a new release with a fix.
1 parent f545712 commit 62ee666

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/test_gpu_examples.py

+6
Original file line numberDiff line numberDiff line change
@@ -3423,6 +3423,12 @@ def test_causal_lm_training_awq(self):
34233423
assert trainer.state.log_history[-1]["train_loss"] is not None
34243424

34253425
@pytest.mark.multi_gpu_tests
3426+
# TODO remove marker if/once issue is resolved, most likely requiring a fix in AutoAWQ:
3427+
# https://github.com/casper-hansen/AutoAWQ/issues/754
3428+
@pytest.mark.xfail(
3429+
reason="Multi-GPU test currently not working with AutoAWQ and PyTorch 2.7",
3430+
strict=True,
3431+
)
34263432
@require_torch_multi_gpu
34273433
def test_causal_lm_training_multi_gpu(self):
34283434
r"""

0 commit comments

Comments
 (0)