fixup

wsmoses · wsmoses · commit 4c87214e07db · 2025-01-15T09:18:46.000-06:00
diff --git a/deps/ReactantExtra/WORKSPACE b/deps/ReactantExtra/WORKSPACE
@@ -9,7 +9,7 @@ http_archive(
     urls = ["https://github.com/wsmoses/nsync/archive/{commit}.tar.gz".format(commit = NSYNC_COMMIT)],
 )
 
-ENZYMEXLA_COMMIT = "1d9cab766d5c35646da956a3b32aff9d61315f90"
+ENZYMEXLA_COMMIT = "64a1c283072d4ce4eb319c69b32a6f3c68f30cbe"
 ENZYMEXLA_SHA256 = ""
 
 http_archive(
diff --git a/ext/ReactantCUDAExt.jl b/ext/ReactantCUDAExt.jl
@@ -656,6 +656,7 @@ Reactant.@reactant_overlay @noinline function (func::LLVMFunc{F,tt})(
         fn=MLIR.IR.FlatSymbolRefAttribute(sym_name),
         output_operand_aliases=MLIR.IR.Attribute(output_operand_aliases),
     )
+    @show string(call), typeof(func.f), collect(map(typeof, args))
 
     argidx = 1
     for arg in values(seen)
diff --git a/src/Compiler.jl b/src/Compiler.jl
@@ -387,7 +387,7 @@ function compile_mlir!(mod, f, args; optimize::Union{Bool,Symbol}=true, no_nan::
     end
     kern = "lower-kernel{run_init=true toolkitPath=$toolkit cuLaunchKernelPtr=$(cuLaunch[]) cuModuleLoadDataPtr=$(cuModule[]) cuModuleGetFunctionPtr=$(cuFunc[])},symbol-dce"
 
-    opt_passes = optimization_passes(; no_nan)
+    opt_passes = optimization_passes(; no_nan, sroa=true)
     opt_passes2 = optimization_passes(; no_nan, sroa=false)
 
     if optimize === :all

Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@ http_archive(`
`9`	`9`	`urls = ["https://github.com/wsmoses/nsync/archive/{commit}.tar.gz".format(commit = NSYNC_COMMIT)],`
`10`	`10`	`)`
`11`	`11`
`12`		`-ENZYMEXLA_COMMIT = "1d9cab766d5c35646da956a3b32aff9d61315f90"`
	`12`	`+ENZYMEXLA_COMMIT = "64a1c283072d4ce4eb319c69b32a6f3c68f30cbe"`
`13`	`13`	`ENZYMEXLA_SHA256 = ""`
`14`	`14`
`15`	`15`	`http_archive(`
Original file line number	Diff line number	Diff line change
`@@ -656,6 +656,7 @@ Reactant.@reactant_overlay @noinline function (func::LLVMFunc{F,tt})(`
`656`	`656`	`fn=MLIR.IR.FlatSymbolRefAttribute(sym_name),`
`657`	`657`	`output_operand_aliases=MLIR.IR.Attribute(output_operand_aliases),`
`658`	`658`	`)`
	`659`	`+ @show string(call), typeof(func.f), collect(map(typeof, args))`
`659`	`660`
`660`	`661`	`argidx = 1`
`661`	`662`	`for arg in values(seen)`