|
1 | | -# Copyright 2024 Arm Limited and/or its affiliates. |
| 1 | +# Copyright 2024-2025 Arm Limited and/or its affiliates. |
2 | 2 | # All rights reserved. |
3 | 3 | # |
4 | 4 | # This source code is licensed under the BSD-style license found in the |
@@ -101,7 +101,6 @@ def _test_split_ethosu_BI_pipeline( |
101 | 101 | ) |
102 | 102 | .quantize() |
103 | 103 | .export() |
104 | | - .check(["torch.ops.aten.split.Tensor"]) |
105 | 104 | .to_edge() |
106 | 105 | .partition() |
107 | 106 | .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): |
129 | 128 | def test_split_tosa_BI(self, test_data: test_data_t): |
130 | 129 | self._test_split_tosa_BI_pipeline(self.Split(), test_data) |
131 | 130 |
|
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) |
135 | 132 | def test_split_u55_BI(self, test_data: test_data_t): |
136 | 133 | self._test_split_ethosu_BI_pipeline( |
137 | 134 | common.get_u55_compile_spec(), self.Split(), test_data |
138 | 135 | ) |
139 | 136 |
|
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) |
151 | 138 | def test_split_u85_BI(self, test_data: test_data_t): |
152 | 139 | self._test_split_ethosu_BI_pipeline( |
153 | 140 | common.get_u85_compile_spec(), self.Split(), test_data |
154 | 141 | ) |
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