Skip to content

Commit 4ebc9c0

Browse files
authored
solve the test issue (#2432)
bring numeric debug test back (#2432) Summary: torchao numeric debug infra have to rely on pytorch 2.8 nightly and beyond; update the test decorator to skip the tests on lower version pytorch Differential Revision: D77191199
1 parent faf788a commit 4ebc9c0

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

test/quantization/pt2e/test_numeric_debugger.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,14 +18,15 @@
1818
prepare_for_propagation_comparison,
1919
)
2020
from torchao.testing.pt2e.utils import PT2ENumericDebuggerTestCase
21-
from torchao.utils import TORCH_VERSION_AT_LEAST_2_7
21+
from torchao.utils import TORCH_VERSION_AT_LEAST_2_8
2222

23-
if TORCH_VERSION_AT_LEAST_2_7:
23+
if TORCH_VERSION_AT_LEAST_2_8:
2424
from torch.export import export_for_training
2525

2626

27-
@unittest.skip("skip for now, need to fix")
28-
@unittest.skipIf(not TORCH_VERSION_AT_LEAST_2_7, "Requires torch 2.7+")
27+
@unittest.skipIf(
28+
not TORCH_VERSION_AT_LEAST_2_8, "Requires torch 2.8 and above, including nightly"
29+
)
2930
@unittest.skipIf(IS_WINDOWS, "Windows not yet supported for torch.compile")
3031
class TestNumericDebuggerInfra(PT2ENumericDebuggerTestCase):
3132
@unittest.skip(

0 commit comments

Comments
 (0)