Skip to content

Commit 435931e

Browse files
metascroyfacebook-github-bot
authored andcommitted
Fix broken fbtest
Summary: Fix broken internal test Differential Revision: D79837913
1 parent afdbb85 commit 435931e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

backends/apple/coreml/test/test_torch_ops.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,7 @@ 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(not hasattr(torch.version, "git_version"), "Enable in fbcode once D79658061 lands")
170171
def test_dequantize_codebook_linear(self):
171172
model, example_inputs = self._get_test_model()
172173
quantize_(
@@ -194,6 +195,7 @@ def test_dequantize_codebook_linear(self):
194195
et_prog = delegated_program.to_executorch()
195196
self._compare_outputs(et_prog, model, example_inputs)
196197

198+
@unittest.skipIf(not hasattr(torch.version, "git_version"), "Enable in fbcode once D79658061 lands")
197199
def test_dequantize_codebook_embedding(self):
198200
model, example_inputs = self._get_test_model()
199201
quantize_(

0 commit comments

Comments
 (0)