Adapt to CompilerCaching 0.3

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"

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

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

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")