Arm backend: Promote InsertDataLayoutCastsPass up pass list

tom-arm · tom-arm · commit 09c0efba2673 · 2026-05-06T14:36:16.000+01:00
* Move above Aten -&gt; TOSA transforms to ensure dtypes are valid

Signed-off-by: Tom Allsop &lt;tom.allsop@arm.com&gt;
Change-Id: I021289f51f75e1a303050d0ffcf3b939f67ee88f
diff --git a/backends/arm/_passes/arm_pass_manager.py b/backends/arm/_passes/arm_pass_manager.py
@@ -526,6 +526,7 @@ def _tosa_pipeline(
                 DecomposeSumPass(),
                 InsertTableOpsPass(exported_program),
                 RemoveNoopPass(),
+                InsertDataLayoutCastsPass(),
             ]
         )
 
@@ -558,7 +559,6 @@ def _tosa_pipeline(
                 EnsureUniqueOutputNodesPass(),
                 RemoveNoopPass(),
                 InsertRescalePass(),
-                InsertDataLayoutCastsPass(),
             ]
         )
 

Original file line number	Diff line number	Diff line change
`@@ -526,6 +526,7 @@ def _tosa_pipeline(`
`526`	`526`	`DecomposeSumPass(),`
`527`	`527`	`InsertTableOpsPass(exported_program),`
`528`	`528`	`RemoveNoopPass(),`
	`529`	`+ InsertDataLayoutCastsPass(),`
`529`	`530`	`]`
`530`	`531`	`)`
`531`	`532`
`@@ -558,7 +559,6 @@ def _tosa_pipeline(`
`558`	`559`	`EnsureUniqueOutputNodesPass(),`
`559`	`560`	`RemoveNoopPass(),`
`560`	`561`	`InsertRescalePass(),`
`561`		`- InsertDataLayoutCastsPass(),`
`562`	`562`	`]`
`563`	`563`	`)`
`564`	`564`