Arm backend: Promote InsertDataLayoutCastsPass up pass list (pytorch#19332)

tom-arm · web-flow · commit 29761c82020b · 2026-05-11T15:46:11.000+01:00
* Move above Aten -> TOSA transforms to ensure dtypes are valid Change-Id: I021289f51f75e1a303050d0ffcf3b939f67ee88f cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils @Sebastian-Larsson @robell Signed-off-by: Tom Allsop <tom.allsop@arm.com>
diff --git a/backends/arm/_passes/arm_pass_manager.py b/backends/arm/_passes/arm_pass_manager.py
@@ -524,6 +524,7 @@ def _tosa_pipeline(
                 DecomposeSumPass(),
                 InsertTableOpsPass(exported_program),
                 RemoveNoopPass(),
+                InsertDataLayoutCastsPass(),
             ]
         )
 
@@ -556,7 +557,6 @@ def _tosa_pipeline(
                 EnsureUniqueOutputNodesPass(),
                 RemoveNoopPass(),
                 InsertRescalePass(),
-                InsertDataLayoutCastsPass(),
             ]
         )
 

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