Commit 62e9c59
Disable _sort_outputs in TOSA backend (MLETORCH-1266)
Summary:
Disables the _sort_outputs call in the TOSA backend lowering.
The _sort_outputs function reorders TOSA flatbuffer outputs, but the
runner reads output shapes from original_module (a pre-preprocess copy),
creating a mismatch between expected and actual output order.
The TOSA reference model preserves flatbuffer order, so no sorting is
actually needed. This change prevents the mismatch issue.
When output_order_workaround is enabled, the code now logs a debug
message indicating the option is set but _sort_outputs is disabled.
Differential Revision: D920590441 parent 691ca29 commit 62e9c59
1 file changed
Lines changed: 9 additions & 5 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
21 | 21 | | |
22 | 22 | | |
23 | 23 | | |
24 | | - | |
25 | 24 | | |
26 | 25 | | |
27 | 26 | | |
| |||
347 | 346 | | |
348 | 347 | | |
349 | 348 | | |
| 349 | + | |
| 350 | + | |
| 351 | + | |
| 352 | + | |
| 353 | + | |
350 | 354 | | |
351 | | - | |
352 | | - | |
353 | | - | |
354 | | - | |
| 355 | + | |
| 356 | + | |
| 357 | + | |
| 358 | + | |
355 | 359 | | |
356 | 360 | | |
357 | 361 | | |
| |||
0 commit comments