From 6e3153b8c4b986b335ff32a29323eff76cf427eb Mon Sep 17 00:00:00 2001 From: jainapurva Date: Thu, 14 Nov 2024 16:37:23 -0800 Subject: [PATCH] skip test in fbcode --- test/quantization/test_marlin_qqq.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/test/quantization/test_marlin_qqq.py b/test/quantization/test_marlin_qqq.py index c020b958f..ebde8f716 100644 --- a/test/quantization/test_marlin_qqq.py +++ b/test/quantization/test_marlin_qqq.py @@ -18,7 +18,11 @@ MappingType, choose_qparams_and_quantize_affine_qqq, ) -from torchao.utils import TORCH_VERSION_AT_LEAST_2_5 +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") class MarlinQQQ(TestCase):