feat: new batching passes [skip ci]

avik-pal · avik-pal · commit 935e9adf16a6 · 2025-08-22T23:18:49.000-05:00
diff --git a/deps/ReactantExtra/WORKSPACE b/deps/ReactantExtra/WORKSPACE
@@ -4,7 +4,7 @@ NSYNC_COMMIT = "82b118aa7ace3132e517e2c467f8732978cf4023"
 
 NSYNC_SHA256 = ""
 
-ENZYMEXLA_COMMIT = "d33fe7cb02c450334c8b9e1fda8f48e671a1d891"
+ENZYMEXLA_COMMIT = "b3b62ef98eec2b0979871a4506b85e60af33a812"
 
 ENZYMEXLA_SHA256 = ""
 
diff --git a/src/Compiler.jl b/src/Compiler.jl
@@ -910,6 +910,9 @@ function optimization_passes(
         "concat_insert_dim_dot_general",
         "concat_insert_dim_gather",
         "concat_insert_dim_iota",
+        "concat_insert_dim_reduce",
+        "concat_insert_dim_sort",
+        "concat_insert_dim_reduce_window",
     ]
 
     if !compile_options.disable_scatter_gather_optimization_passes

Original file line number	Diff line number	Diff line change
`@@ -910,6 +910,9 @@ function optimization_passes(`
`910`	`910`	`"concat_insert_dim_dot_general",`
`911`	`911`	`"concat_insert_dim_gather",`
`912`	`912`	`"concat_insert_dim_iota",`
	`913`	`+ "concat_insert_dim_reduce",`
	`914`	`+ "concat_insert_dim_sort",`
	`915`	`+ "concat_insert_dim_reduce_window",`
`913`	`916`	`]`
`914`	`917`
`915`	`918`	`if !compile_options.disable_scatter_gather_optimization_passes`