Skip to content

Commit 0dba025

Browse files
authored
Remove incorrectly xfailing split tests (#7648)
Test failed due to checking for split op when the test contained a split_with_sizes op. Remove check. Signed-off-by: Erik Lundell <[email protected]>
1 parent a4def9f commit 0dba025

File tree

1 file changed

+3
-23
lines changed

1 file changed

+3
-23
lines changed

backends/arm/test/ops/test_split.py

+3-23
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2024 Arm Limited and/or its affiliates.
1+
# Copyright 2024-2025 Arm Limited and/or its affiliates.
22
# All rights reserved.
33
#
44
# This source code is licensed under the BSD-style license found in the
@@ -101,7 +101,6 @@ def _test_split_ethosu_BI_pipeline(
101101
)
102102
.quantize()
103103
.export()
104-
.check(["torch.ops.aten.split.Tensor"])
105104
.to_edge()
106105
.partition()
107106
.check_count({"torch.ops.higher_order.executorch_call_delegate": 1})
@@ -129,33 +128,14 @@ def test_split_two_out_tosa_MI(self, test_data: test_data_t):
129128
def test_split_tosa_BI(self, test_data: test_data_t):
130129
self._test_split_tosa_BI_pipeline(self.Split(), test_data)
131130

132-
@parameterized.expand(
133-
[Split.test_data[0], Split.test_data[1], Split.test_data[2], Split.test_data[4]]
134-
)
131+
@parameterized.expand(Split.test_data)
135132
def test_split_u55_BI(self, test_data: test_data_t):
136133
self._test_split_ethosu_BI_pipeline(
137134
common.get_u55_compile_spec(), self.Split(), test_data
138135
)
139136

140-
# TODO MLETORCH-350
141-
@parameterized.expand([Split.test_data[3], Split.test_data[5]])
142-
@unittest.expectedFailure
143-
def test_split_u55_BI_skip(self, test_data: test_data_t):
144-
self._test_split_ethosu_BI_pipeline(
145-
common.get_u55_compile_spec(), self.Split(), test_data
146-
)
147-
148-
@parameterized.expand(
149-
[Split.test_data[0], Split.test_data[1], Split.test_data[2], Split.test_data[4]]
150-
)
137+
@parameterized.expand(Split.test_data)
151138
def test_split_u85_BI(self, test_data: test_data_t):
152139
self._test_split_ethosu_BI_pipeline(
153140
common.get_u85_compile_spec(), self.Split(), test_data
154141
)
155-
156-
@parameterized.expand([Split.test_data[3], Split.test_data[5]])
157-
@unittest.expectedFailure
158-
def test_split_u85_BI_skip(self, test_data: test_data_t):
159-
self._test_split_ethosu_BI_pipeline(
160-
common.get_u85_compile_spec(), self.Split(), test_data
161-
)

0 commit comments

Comments
 (0)