Skip to content

Commit bc874fc

Browse files
committed
add nightly marker for torch.compile test
1 parent e9fccb6 commit bc874fc

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/quantization/torchao/test_torchao.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -365,6 +365,7 @@ def test_training(self):
365365
self.assertTrue(module.adapter[1].weight.grad is not None)
366366
self.assertTrue(module.adapter[1].weight.grad.norm().item() > 0)
367367

368+
@nightly
368369
def test_torch_compile(self):
369370
r"""Test that verifies if torch.compile works with torchao quantization."""
370371
quantization_config = TorchAoConfig("int8_weight_only")

0 commit comments

Comments
 (0)