Apply suggestions from code review

EnzymeAD · Jun 12, 2024 · 4bb8bd3 · 4bb8bd3
1 parent 488e715
commit 4bb8bd3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/compiler/interpreter.jl b/src/compiler/interpreter.jl
@@ -63,9 +63,9 @@ Core.Compiler.OptimizationParams(interp::EnzymeInterpreter) = interp.opt_params
 get_inference_world(interp::EnzymeInterpreter) = interp.world
 Core.Compiler.get_inference_cache(interp::EnzymeInterpreter) = interp.local_cache
 @static if HAS_INTEGRATED_CACHE
-    Core.Compiler.cache_owner(interp::EnzymeInterpreter) = interp.cache_or_token
+    Core.Compiler.cache_owner(interp::EnzymeInterpreter) = interp.token
 else
-    Core.Compiler.code_cache(interp::EnzymeInterpreter) = WorldView(interp.cache_or_token, interp.world)
+    Core.Compiler.code_cache(interp::EnzymeInterpreter) = WorldView(interp.cache, interp.world)
 end
 
 # No need to do any locking since we're not putting our results into the runtime cache