|
1 | | -# Copyright 2024 NXP |
| 1 | +# Copyright 2024-2025 NXP |
2 | 2 | # |
3 | 3 | # This source code is licensed under the BSD-style license found in the |
4 | 4 | # LICENSE file in the root directory of this source tree. |
@@ -165,7 +165,7 @@ def test_delegating_format_related_transpose_operators__supported_case(mocker): |
165 | 165 | # Make sure the output channels (channels for the trailing Transpose), and the last input dimension (channels for |
166 | 166 | # the leading Transpose) are multiples of `num_macs``. |
167 | 167 |
|
168 | | - num_macs = NeutronTargetSpec("imxrt700", "SDK_25_09").get_num_macs() |
| 168 | + num_macs = NeutronTargetSpec("imxrt700", "SDK_25_12").get_num_macs() |
169 | 169 | model = Conv2dModule( |
170 | 170 | in_channels=num_macs, out_channels=num_macs, padding=1, stride=1 |
171 | 171 | ) |
@@ -222,7 +222,7 @@ def test_delegating_format_related_transpose_operators__supported_case(mocker): |
222 | 222 | def test_delegating_format_related_transpose_operators__supported_output__unsupported_input( |
223 | 223 | mocker, |
224 | 224 | ): |
225 | | - num_macs = NeutronTargetSpec("imxrt700", "SDK_25_09").get_num_macs() |
| 225 | + num_macs = NeutronTargetSpec("imxrt700", "SDK_25_12").get_num_macs() |
226 | 226 | model = Conv2dModule( |
227 | 227 | in_channels=num_macs, |
228 | 228 | out_channels=num_macs, # The output `Transpose` will be supported. |
@@ -278,7 +278,7 @@ def test_delegating_format_related_transpose_operators__supported_output__unsupp |
278 | 278 | def test_delegating_format_related_transpose_operators__supported_input__unsupported_output( |
279 | 279 | mocker, |
280 | 280 | ): |
281 | | - num_macs = NeutronTargetSpec("imxrt700", "SDK_25_09").get_num_macs() |
| 281 | + num_macs = NeutronTargetSpec("imxrt700", "SDK_25_12").get_num_macs() |
282 | 282 | model = Conv2dModule( |
283 | 283 | in_channels=num_macs, |
284 | 284 | out_channels=3, # The output `Transpose` will NOT be supported. |
|
0 commit comments