|
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