From 2956b4d5377a32fea4cc9c36a2d1d2f7317fcf45 Mon Sep 17 00:00:00 2001 From: Billy Moses Date: Tue, 3 Dec 2024 08:29:12 -0600 Subject: [PATCH] fix --- src/compiler.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/compiler.jl b/src/compiler.jl index 13fc301b0b..10dc9842d1 100644 --- a/src/compiler.jl +++ b/src/compiler.jl @@ -5316,10 +5316,10 @@ const cache_lock = ReentrantLock() asm = _thunk(job) obj = _link(job, asm...) if obj.adjoint isa Ptr{Nothing} - autodiff_cache[obj.adjoint] = (asm[2], asm[1]) + autodiff_cache[obj.adjoint] = (asm[2], asm[5]) end if obj.primal isa Ptr{Nothing} - autodiff_cache[obj.primal] = (asm[3], asm[1]) + autodiff_cache[obj.primal] = (asm[3], asm[5]) end cache[key] = obj end