diag opt passes (#1510)

wsmoses · web-flow · commit 70f4ab326a2c · 2025-07-31T11:50:34.000-05:00
diff --git a/src/Compiler.jl b/src/Compiler.jl
@@ -889,6 +889,8 @@ function optimization_passes(
         "neg_div_const_simplify",
         "reshape_deletions_broadcast_in_dim_simplify",
         "reshape_insertions_broadcast_in_dim_simplify",
+        "dot_general_reshape",
+        "diagonal_tensor_dot_general_rewrite",
     ]
 
     if !compile_options.disable_scatter_gather_optimization_passes

Original file line number	Diff line number	Diff line change
`@@ -889,6 +889,8 @@ function optimization_passes(`
`889`	`889`	`"neg_div_const_simplify",`
`890`	`890`	`"reshape_deletions_broadcast_in_dim_simplify",`
`891`	`891`	`"reshape_insertions_broadcast_in_dim_simplify",`
	`892`	`+ "dot_general_reshape",`
	`893`	`+ "diagonal_tensor_dot_general_rewrite",`
`892`	`894`	`]`
`893`	`895`
`894`	`896`	`if !compile_options.disable_scatter_gather_optimization_passes`