Skip to content

Commit 0f70a5d

Browse files
authored
Fix broken fbtest
Differential Revision: D79837913 Pull Request resolved: #13203
1 parent f7ddbde commit 0f70a5d

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

backends/apple/coreml/test/test_torch_ops.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,10 @@ def test_dequantize_affine_c8w_embedding_b4w_linear(self):
167167
et_prog = delegated_program.to_executorch()
168168
self._compare_outputs(et_prog, model, example_inputs)
169169

170+
@unittest.skipIf(
171+
not hasattr(torch.version, "git_version"),
172+
"Enable in fbcode once D79658061 lands",
173+
)
170174
def test_dequantize_codebook_linear(self):
171175
model, example_inputs = self._get_test_model()
172176
quantize_(
@@ -194,6 +198,10 @@ def test_dequantize_codebook_linear(self):
194198
et_prog = delegated_program.to_executorch()
195199
self._compare_outputs(et_prog, model, example_inputs)
196200

201+
@unittest.skipIf(
202+
not hasattr(torch.version, "git_version"),
203+
"Enable in fbcode once D79658061 lands",
204+
)
197205
def test_dequantize_codebook_embedding(self):
198206
model, example_inputs = self._get_test_model()
199207
quantize_(

0 commit comments

Comments
 (0)