fix

Author: wsmoses

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 = "1b473e8e77850ece61bb0e85b152b95cb6a70be0"
+ENZYMEXLA_COMMIT = "1d9cab766d5c35646da956a3b32aff9d61315f90"
 ENZYMEXLA_SHA256 = ""
 
 http_archive(
@@ -95,7 +95,7 @@ LLVM_TARGETS = select({
 }) + ["AArch64", "X86", "ARM"]
 
 # Uncomment these lines to use a custom LLVM commit
-LLVM_COMMIT = "9b4bf06be33f0fe6a4c487bb9244d8c0f6acab3f"
+LLVM_COMMIT = "b39c5cb6977f35ad727d86b2dd6232099734ffd3"
 LLVM_SHA256 = ""
 http_archive(
     name = "llvm-raw",

ext/ReactantCUDAExt.jl

@@ -576,6 +576,20 @@ Reactant.@reactant_overlay @noinline function (func::LLVMFunc{F,tt})(
         push!(restys, MLIR.IR.type(arg))
         push!(mlir_args, arg)
 
+        push!(
+            aliases,
+            MLIR.IR.Attribute(
+                MLIR.API.stablehloOutputOperandAliasGet(
+                    MLIR.IR.context(),
+                    length(wrapper_tys) == 1 ? 0 : 1,
+                    length(wrapper_tys) == 1 ? C_NULL : Ref{Int64}(argidx - 1),
+                    argidx - 1,
+                    0,
+                    C_NULL,
+                ),
+            ),
+        )
+
         for p in paths
             if p[1] !== kernelargsym
                 continue
@@ -600,20 +614,6 @@ Reactant.@reactant_overlay @noinline function (func::LLVMFunc{F,tt})(
                 )
                 MLIR.Dialects.llvm.store(MLIR.IR.argument(wrapbody, argidx), ptr)
             end
-
-            push!(
-                aliases,
-                MLIR.IR.Attribute(
-                    MLIR.API.stablehloOutputOperandAliasGet(
-                        MLIR.IR.context(),
-                        length(wrapper_tys) == 1 ? 0 : 1,
-                        length(wrapper_tys) == 1 ? C_NULL : Ref{Int64}(argidx - 1),
-                        argidx - 1,
-                        0,
-                        C_NULL,
-                    ),
-                ),
-            )
         end
         argidx += 1
     end