diff --git a/test/quantization/test_marlin_qqq.py b/test/quantization/test_marlin_qqq.py index 3531f02f7b..0dcaaf9c8c 100644 --- a/test/quantization/test_marlin_qqq.py +++ b/test/quantization/test_marlin_qqq.py @@ -21,10 +21,7 @@ from torchao.utils import TORCH_VERSION_AT_LEAST_2_5, is_fbcode import unittest -if is_fbcode(): - unittest.skip("Skipping the test in fbcode since we don't have TARGET file for kernels") - - +@unittest.skipIf(is_fbcode(), "Skipping the test in fbcode since we don't have TARGET file for kernels") class TestMarlinQQQ(TestCase): def setUp(self): super().setUp()