From f2070bc73a11aafee156ce988b7e9a3c30c17ba4 Mon Sep 17 00:00:00 2001 From: Apurva Jain Date: Fri, 15 Nov 2024 14:08:46 -0800 Subject: [PATCH] Fix Marlin QQQ ops test with unittest (#1294) Summary: Pull Request resolved: https://github.com/pytorch/ao/pull/1294 Skipping test_marlin_qqq in fbcode Reviewed By: jerryzh168 Differential Revision: D66031641 --- test/quantization/test_marlin_qqq.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) 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()