We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e9fccb6 commit bc874fcCopy full SHA for bc874fc
tests/quantization/torchao/test_torchao.py
@@ -365,6 +365,7 @@ def test_training(self):
365
self.assertTrue(module.adapter[1].weight.grad is not None)
366
self.assertTrue(module.adapter[1].weight.grad.norm().item() > 0)
367
368
+ @nightly
369
def test_torch_compile(self):
370
r"""Test that verifies if torch.compile works with torchao quantization."""
371
quantization_config = TorchAoConfig("int8_weight_only")
0 commit comments