Fix decayaddr to handle stores into the decayed addr (#1109)

EnzymeAD · Oct 25, 2023 · 558abc5 · 558abc5
1 parent c6b265b
commit 558abc5
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/compiler/optimize.jl b/src/compiler/optimize.jl
@@ -269,7 +269,16 @@ function fix_decayaddr!(mod::LLVM.Module)
             temp = nothing
             for u in LLVM.uses(inst)
                 st = LLVM.user(u)
-                if !isa(st, LLVM.CallInst)
+					 # Storing _into_ the decay addr is okay
+					 # we just cannot store the decayed addr into
+					 # somewhere
+					 if isa(st, LLVM.StoreInst)
+					    if operands(st)[2] == inst
+							 LLVM.API.LLVMSetOperand(st, 2-1, operands(inst)[1])
+							 continue
+						 end
+					 end
+					 if !isa(st, LLVM.CallInst)
 						  bt = GPUCompiler.backtrace(st)
 						  msg = sprint() do io::IO
 							  println(io, string(f))