adjust to recent binding partition changes (#61)

Author: aviatesk

CassetteBase/Project.toml

@@ -1,7 +1,7 @@
 name = "CassetteBase"
 uuid = "6dd3e646-b1c5-42c7-94be-00277fa12e22"
 authors = ["Shuhei Kadowaki <[email protected]>"]
-version = "0.1.0"
+version = "0.1.1"
 
 [compat]
 julia = "1.10"

CassetteBase/src/CassetteBase.jl

@@ -1,10 +1,12 @@
 module CassetteBase
 
-export cassette_transform!, generate_internalerr_ex, generate_lambda_ex
+export SourceType, cassette_transform!, generate_internalerr_ex, generate_lambda_ex
 
 using Core.IR
 using Core: SimpleVector
 
+const SourceType = @static VERSION ≥ v"1.12.0-DEV.1968" ? Method : LineNumberNode
+
 function cassette_transform!(src::CodeInfo, mi::MethodInstance, nargs::Int,
                              selfname::Symbol, fargsname::Symbol)
     method = mi.def::Method
@@ -146,7 +148,7 @@ function Base.showerror(io::IO, err::CassetteInternalError)
 end
 
 function generate_internalerr_ex(err, bt::Vector, context::Symbol,
-                                 world::UInt, source::LineNumberNode,
+                                 world::UInt, source::SourceType,
                                  argnames::SimpleVector, spnames::SimpleVector,
                                  metadata=nothing)
     @nospecialize err metadata
@@ -155,7 +157,7 @@ function generate_internalerr_ex(err, bt::Vector, context::Symbol,
     return generate_lambda_ex(world, source, argnames, spnames, throw_ex)
 end
 
-function generate_lambda_ex(world::UInt, source::LineNumberNode,
+function generate_lambda_ex(world::UInt, source::SourceType,
                             argnames::SimpleVector, spnames::SimpleVector,
                             body::Expr)
     stub = Core.GeneratedFunctionStub(identity, argnames, spnames)

CassetteBase/test/test_basic.jl

@@ -2,12 +2,12 @@ module test_basic
 
 using Test, CassetteBase
 
-struct BasicGenerator
+struct BasicGenerator <: (@static isdefined(Core, :CachedGenerator) ? Core.CachedGenerator : Any)
     selfname::Symbol
     fargsname::Symbol
     raise::Bool
 end
-function (generator::BasicGenerator)(world::UInt, source::LineNumberNode, passtype, fargtypes)
+function (generator::BasicGenerator)(world::UInt, source::SourceType, passtype, fargtypes)
     @nospecialize passtype fargtypes
     (; selfname, fargsname, raise) = generator
     try
@@ -21,7 +21,7 @@ function (generator::BasicGenerator)(world::UInt, source::LineNumberNode, passty
             #=metadata=#(; world, source, passtype, fargtypes))
     end
 end
-function generate_basic_src(world::UInt, source::LineNumberNode, passtype, fargtypes,
+function generate_basic_src(world::UInt, source::SourceType, passtype, fargtypes,
                             selfname::Symbol, fargsname::Symbol; raise::Bool)
     @nospecialize passtype fargtypes
     tt = Base.to_tuple_type(fargtypes)

Project.toml

@@ -1,7 +1,7 @@
 name = "CassetteOverlay"
 uuid = "d78b62d4-37fa-4a6f-acd8-2f19986eb9ee"
 authors = ["JuliaHub, Inc. and other contributors"]
-version = "0.2.1"
+version = "0.2.2"
 
 [deps]
 CassetteBase = "6dd3e646-b1c5-42c7-94be-00277fa12e22"

src/CassetteOverlay.jl

@@ -34,7 +34,7 @@ struct CassetteOverlayGenerator <: (@static isdefined(Core, :CachedGenerator) ?
     selfname::Symbol
     fargsname::Symbol
 end
-function (generator::CassetteOverlayGenerator)(world::UInt, source::LineNumberNode, passtype, fargtypes)
+function (generator::CassetteOverlayGenerator)(world::UInt, source::SourceType, passtype, fargtypes)
     @nospecialize passtype fargtypes
     (; selfname, fargsname) = generator
     try
@@ -48,7 +48,7 @@ function (generator::CassetteOverlayGenerator)(world::UInt, source::LineNumberNo
     end
 end
 
-function generate_overlay_src(world::UInt, #=source=#::LineNumberNode, passtype, fargtypes,
+function generate_overlay_src(world::UInt, #=source=#::SourceType, passtype, fargtypes,
                               selfname::Symbol, fargsname::Symbol)
     @nospecialize passtype fargtypes
     tt = Base.to_tuple_type(fargtypes)
@@ -135,23 +135,29 @@ macro overlaypass(args...)
 
     push!(topblk.args, :(return $retval))
 
-    return topblk
+    # attach :latestworld if necessary (N.B. adding it the :toplevel block doesn't work)
+    @static if VERSION ≥ v"1.12.0-DEV.1662"
+        return Expr(:block, Expr(:(=), :pass, topblk), Expr(:latestworld), :pass)
+    else
+        return topblk
+    end
 end
 
 abstract type AbstractBindingOverlay{M, S} <: OverlayPass; end
 function methodtable(::Type{<:AbstractBindingOverlay{M, S}}) where {M, S}
     if M === nothing
         return nothing
     end
-    @assert isconst(M, S)
-    return getglobal(M, S)::MethodTable
+    @assert @invokelatest isconst(M, S)
+    mt = @invokelatest getglobal(M, S)
+    return mt::MethodTable
 end
 @overlaypass AbstractBindingOverlay nothing
 
 struct Overlay{M, S} <: AbstractBindingOverlay{M, S}; end
 function Overlay(mt::MethodTable)
-    @assert isconst(mt.module, mt.name)
-    @assert getglobal(mt.module, mt.name) === mt
+    @assert @invokelatest isconst(mt.module, mt.name)
+    @assert mt === @invokelatest getglobal(mt.module, mt.name)
     return Overlay{mt.module, mt.name}()
 end