diff --git a/backends/apple/coreml/test/test_torch_ops.py b/backends/apple/coreml/test/test_torch_ops.py index 89eab1a8b00..67bc8be197d 100644 --- a/backends/apple/coreml/test/test_torch_ops.py +++ b/backends/apple/coreml/test/test_torch_ops.py @@ -167,6 +167,10 @@ def test_dequantize_affine_c8w_embedding_b4w_linear(self): et_prog = delegated_program.to_executorch() self._compare_outputs(et_prog, model, example_inputs) + @unittest.skipIf( + not hasattr(torch.version, "git_version"), + "Enable in fbcode once D79658061 lands", + ) def test_dequantize_codebook_linear(self): model, example_inputs = self._get_test_model() quantize_( @@ -194,6 +198,10 @@ def test_dequantize_codebook_linear(self): et_prog = delegated_program.to_executorch() self._compare_outputs(et_prog, model, example_inputs) + @unittest.skipIf( + not hasattr(torch.version, "git_version"), + "Enable in fbcode once D79658061 lands", + ) def test_dequantize_codebook_embedding(self): model, example_inputs = self._get_test_model() quantize_(