Skip to content

Commit 13408b9

Browse files
zingofacebook-github-bot
authored andcommitted
Arm backend: Avoid failing sigmoid unit tests after Pytorch 2.6 update (#5843)
Summary: The Pytorch 2.5 -> 2.6 updated caused tests using sigmoid together with add in quantize mode to start failing. Until the problem is solved lets avoid the tests to not stop CI jobs. See #5832 Pull Request resolved: #5843 Reviewed By: digantdesai Differential Revision: D63832203 Pulled By: mergennachin fbshipit-source-id: 746177487e71ac17c9f6a522f68508298759c1db
1 parent 0e5b92d commit 13408b9

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

backends/arm/test/ops/test_sigmoid.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,12 +152,32 @@ def test_sigmoid_tosa_MI(
152152
def test_sigmoid_tosa_BI(self, test_name: str, test_data: torch.Tensor):
153153
self._test_sigmoid_tosa_BI_pipeline(self.Sigmoid(), (test_data,))
154154

155+
def test_add_sigmoid_tosa_MI(self):
156+
self._test_sigmoid_tosa_MI_pipeline(self.AddSigmoid(), (test_data_suite[0][1],))
157+
158+
@unittest.skip(
159+
reason="Started to fails when PyTorch 2.5->2.6 https://github.com/pytorch/executorch/issues/5832"
160+
)
155161
def test_add_sigmoid_tosa_BI(self):
156162
self._test_sigmoid_tosa_BI_pipeline(self.AddSigmoid(), (test_data_suite[0][1],))
157163

164+
def test_sigmoid_add_tosa_MI(self):
165+
self._test_sigmoid_tosa_MI_pipeline(self.SigmoidAdd(), (test_data_suite[0][1],))
166+
167+
@unittest.skip(
168+
reason="Started to fails when PyTorch 2.5->2.6 https://github.com/pytorch/executorch/issues/5832"
169+
)
158170
def test_sigmoid_add_tosa_BI(self):
159171
self._test_sigmoid_tosa_BI_pipeline(self.SigmoidAdd(), (test_data_suite[0][1],))
160172

173+
def test_sigmoid_add_sigmoid_tosa_MI(self):
174+
self._test_sigmoid_tosa_MI_pipeline(
175+
self.SigmoidAddSigmoid(), (test_data_suite[4][1], test_data_suite[3][1])
176+
)
177+
178+
@unittest.skip(
179+
reason="Started to fails when PyTorch 2.5->2.6 https://github.com/pytorch/executorch/issues/5832"
180+
)
161181
def test_sigmoid_add_sigmoid_tosa_BI(self):
162182
self._test_sigmoid_tosa_BI_pipeline(
163183
self.SigmoidAddSigmoid(), (test_data_suite[4][1], test_data_suite[3][1])

0 commit comments

Comments
 (0)