From 9a7911bcf6d055f16d5dbdc9c39e51b35b0048b8 Mon Sep 17 00:00:00 2001
From: Tim Besard <tim.besard@gmail.com>
Date: Tue, 14 Jul 2026 15:48:11 +0200
Subject: [PATCH] Adapt to CompilerCaching 0.3

---
 Project.toml                | 2 +-
 src/compiler/driver.jl      | 4 ++--
 src/compiler/interpreter.jl | 4 ++--
 src/cuTile.jl               | 2 +-
 4 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/Project.toml b/Project.toml
index 749e3cfb..e5317189 100644
--- a/Project.toml
+++ b/Project.toml
@@ -39,7 +39,7 @@ BFloat16s = "0.6"
 CUDACore = "6.1"
 CUDA_Compiler_jll = "0.4"
 CUDA_Tile_jll = "13.3"
-CompilerCaching = "0.2.4"
+CompilerCaching = "0.3.1"
 DLFP8Types = "0.1.0"
 EnumX = "1.0"
 GPUArrays = "11"
diff --git a/src/compiler/driver.jl b/src/compiler/driver.jl
index f4c93ac6..f7885687 100644
--- a/src/compiler/driver.jl
+++ b/src/compiler/driver.jl
@@ -95,8 +95,8 @@ function get_ci(cache::CacheView, mi::Core.MethodInstance;
     ci = get(cache, mi, nothing)
     if ci === nothing
         interp = cuTileInterpreter(cache)
-        typeinf!(cache, interp, mi)
-        ci = get(cache, mi)
+        ci = typeinf!(interp, mi)
+        ci === nothing && error("Inference failed for $mi")
     end
 
     # Run const-prop inference, if needed
diff --git a/src/compiler/interpreter.jl b/src/compiler/interpreter.jl
index c5e03515..9be96289 100644
--- a/src/compiler/interpreter.jl
+++ b/src/compiler/interpreter.jl
@@ -54,8 +54,8 @@ CC.method_table(interp::cuTileInterpreter) = interp.method_table
 CC.lock_mi_inference(::cuTileInterpreter, ::MethodInstance) = nothing
 CC.unlock_mi_inference(::cuTileInterpreter, ::MethodInstance) = nothing
 
-# Setup caching - generates cache_owner and ipo_dataflow_analysis! methods
-@setup_caching cuTileInterpreter.cache
+# Cache partition
+CC.cache_owner(interp::cuTileInterpreter) = interp.cache.owner
 
 # Optimization flags
 CC.may_optimize(::cuTileInterpreter) = true
diff --git a/src/cuTile.jl b/src/cuTile.jl
index 5bc6a17a..d3ae0f13 100644
--- a/src/cuTile.jl
+++ b/src/cuTile.jl
@@ -22,7 +22,7 @@ using EnumX
 public BFloat16
 
 import CompilerCaching
-using CompilerCaching: CacheView, @setup_caching, method_instance, match_method_instance, typeinf!, results, lookup, get_source
+using CompilerCaching: CacheView, method_instance, match_method_instance, typeinf!, results, lookup, get_source
 
 # Shared definitions
 include("shapes.jl")