Skip to content

Commit 5eedd6e

Browse files
committed
[Backend Tester] Skip inf and nan operator tests
ghstack-source-id: c298968 ghstack-comment-id: 3257173725 Pull-Request: #13990
1 parent f941f2d commit 5eedd6e

23 files changed

+47
-0
lines changed

backends/test/suite/operators/test_abs.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88

99

1010
import torch
11+
import unittest
1112
from executorch.backends.test.suite.flow import TestFlow
1213

1314
from executorch.backends.test.suite.operators import (
@@ -45,6 +46,7 @@ def test_abs_shapes(self, flow: TestFlow) -> None:
4546
# 3D tensor
4647
self._test_op(AbsModel(), (torch.randn(3, 4, 5),), flow)
4748

49+
@unittest.skip("NaN and Inf are not enforced for backends.")
4850
def test_abs_edge_cases(self, flow: TestFlow) -> None:
4951
# Test edge cases
5052

backends/test/suite/operators/test_amax.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from typing import List, Optional, Tuple, Union
1010

1111
import torch
12+
import unittest
1213
from executorch.backends.test.suite.flow import TestFlow
1314

1415
from executorch.backends.test.suite.operators import (
@@ -201,6 +202,7 @@ def test_amax_shapes(self, flow: TestFlow) -> None:
201202
flow,
202203
)
203204

205+
@unittest.skip("NaN and Inf are not enforced for backends.")
204206
def test_amax_edge_cases(self, flow: TestFlow) -> None:
205207
x = torch.tensor([[1.0, float("inf"), 3.0], [4.0, 5.0, float("inf")]])
206208
self._test_op(

backends/test/suite/operators/test_amin.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from typing import List, Optional, Tuple, Union
1010

1111
import torch
12+
import unittest
1213
from executorch.backends.test.suite.flow import TestFlow
1314

1415
from executorch.backends.test.suite.operators import (
@@ -203,6 +204,7 @@ def test_amin_shapes(self, flow: TestFlow) -> None:
203204
flow,
204205
)
205206

207+
@unittest.skip("NaN and Inf are not enforced for backends.")
206208
def test_amin_edge_cases(self, flow: TestFlow) -> None:
207209
x = torch.tensor([[1.0, float("-inf"), 3.0], [4.0, 5.0, float("-inf")]])
208210
self._test_op(

backends/test/suite/operators/test_argmax.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from typing import Optional
1010

1111
import torch
12+
import unittest
1213
from executorch.backends.test.suite.flow import TestFlow
1314

1415
from executorch.backends.test.suite.operators import (
@@ -143,6 +144,7 @@ def test_argmax_shapes(self, flow: TestFlow) -> None:
143144
flow,
144145
)
145146

147+
@unittest.skip("NaN and Inf are not enforced for backends.")
146148
def test_argmax_edge_cases(self, flow: TestFlow) -> None:
147149
x = torch.tensor([[1.0, float("inf"), 3.0], [4.0, 5.0, float("inf")]])
148150
self._test_op(

backends/test/suite/operators/test_argmin.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from typing import Optional
1010

1111
import torch
12+
import unittest
1213
from executorch.backends.test.suite.flow import TestFlow
1314

1415
from executorch.backends.test.suite.operators import (
@@ -143,6 +144,7 @@ def test_argmin_shapes(self, flow: TestFlow) -> None:
143144
flow,
144145
)
145146

147+
@unittest.skip("NaN and Inf are not enforced for backends.")
146148
def test_argmin_edge_cases(self, flow: TestFlow) -> None:
147149
x = torch.tensor([[1.0, float("-inf"), 3.0], [4.0, 5.0, float("-inf")]])
148150
self._test_op(

backends/test/suite/operators/test_ceil.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88

99

1010
import torch
11+
import unittest
1112
from executorch.backends.test.suite.flow import TestFlow
1213

1314
from executorch.backends.test.suite.operators import (
@@ -45,6 +46,7 @@ def test_ceil_shapes(self, flow: TestFlow) -> None:
4546
# 3D tensor
4647
self._test_op(CeilModel(), (torch.randn(3, 4, 5),), flow)
4748

49+
@unittest.skip("NaN and Inf are not enforced for backends.")
4850
def test_ceil_edge_cases(self, flow: TestFlow) -> None:
4951
# Test edge cases
5052

backends/test/suite/operators/test_clamp.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88

99

1010
import torch
11+
import unittest
1112
from executorch.backends.test.suite.flow import TestFlow
1213

1314
from executorch.backends.test.suite.operators import (
@@ -56,6 +57,7 @@ def test_clamp_shapes(self, flow: TestFlow) -> None:
5657
# 3D tensor
5758
self._test_op(model, (torch.randn(3, 4, 5),), flow)
5859

60+
@unittest.skip("NaN and Inf are not enforced for backends.")
5961
def test_clamp_edge_cases(self, flow: TestFlow) -> None:
6062
# Test edge cases
6163

backends/test/suite/operators/test_exp.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88

99

1010
import torch
11+
import unittest
1112
from executorch.backends.test.suite.flow import TestFlow
1213

1314
from executorch.backends.test.suite.operators import (
@@ -46,6 +47,7 @@ def test_exp_shapes(self, flow: TestFlow) -> None:
4647
# 3D tensor
4748
self._test_op(ExpModel(), (torch.randn(3, 4, 5),), flow)
4849

50+
@unittest.skip("NaN and Inf are not enforced for backends.")
4951
def test_exp_edge_cases(self, flow: TestFlow) -> None:
5052
# Test edge cases
5153

backends/test/suite/operators/test_floor.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88

99

1010
import torch
11+
import unittest
1112
from executorch.backends.test.suite.flow import TestFlow
1213

1314
from executorch.backends.test.suite.operators import (
@@ -42,6 +43,7 @@ def test_floor_shapes(self, flow: TestFlow) -> None:
4243
# 3D tensor
4344
self._test_op(FloorModel(), (torch.randn(3, 4, 5),), flow)
4445

46+
@unittest.skip("NaN and Inf are not enforced for backends.")
4547
def test_floor_edge_cases(self, flow: TestFlow) -> None:
4648
# Test edge cases
4749

backends/test/suite/operators/test_floor_divide.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
# pyre-unsafe
88

99
import torch
10+
import unittest
1011
from executorch.backends.test.suite.flow import TestFlow
1112

1213
from executorch.backends.test.suite.operators import (
@@ -178,6 +179,7 @@ def test_floor_divide_values(self, flow: TestFlow) -> None:
178179
y = torch.tensor([-2.0]).expand_as(x).clone()
179180
self._test_op(model, (x, y), flow, generate_random_test_inputs=False)
180181

182+
@unittest.skip("NaN and Inf are not enforced for backends.")
181183
def test_floor_divide_edge_cases(self, flow: TestFlow) -> None:
182184
# Test edge cases
183185
model = FloorDivideModel()

0 commit comments

Comments
 (0)