Commit 0d5192c
authored
Cortex-M backend: Refactor ConvertToCortexMPass (pytorch#20070)
Change to use the general AtenToDialectPass structure,
quantized_op_fusion_pass
to be changed similarly in an upcoming PR.
Removes the 1-1 ensure check as it was to restrictive, cortex-m backend
needs to both insert scratch nodes and for BMM an additional transpose
at the time of dialect replacement.
Bonus small fix: Moves yolo import into test to avoid download at test
collection time
---------
Signed-off-by: Adrian Lundell <adrian.lundell@arm.com>1 parent ff2bf9c commit 0d5192c
8 files changed
Lines changed: 634 additions & 628 deletions
File tree
- backends
- cortex_m
- passes
- test/models
- transforms
- test
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
29 | 29 | | |
30 | 30 | | |
31 | 31 | | |
| 32 | + | |
32 | 33 | | |
33 | | - | |
34 | 34 | | |
35 | 35 | | |
36 | 36 | | |
| |||
45 | 45 | | |
46 | 46 | | |
47 | 47 | | |
| 48 | + | |
48 | 49 | | |
49 | 50 | | |
| 51 | + | |
50 | 52 | | |
51 | 53 | | |
52 | 54 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
35 | 35 | | |
36 | 36 | | |
37 | 37 | | |
| 38 | + | |
38 | 39 | | |
39 | | - | |
40 | 40 | | |
41 | 41 | | |
42 | 42 | | |
| |||
0 commit comments