Skip to content

Commit b8543ab

Browse files
metascroyfacebook-github-bot
authored andcommitted
Fix broken fbtest (#13203)
Summary: Pull Request resolved: #13203 Fix broken internal test Reviewed By: kirklandsign Differential Revision: D79837913
1 parent afdbb85 commit b8543ab

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)