fix: loading reactant on 1.12 (#1755)

* fix: loading reactant on 1.12

* chore: add a warning msg for 1.12

* fix: flip recommendation order

* fix: missing is_closure

* test: fix

Author: avik-pal

src/Compiler.jl

@@ -3714,6 +3714,18 @@ struct Thunk{FTy,tag,IsClosure,ArgTypes,ExecTy,DeviceTy,ClientTy,GD,DAM}
     donated_args_mask::DAM
 end
 
+for fn in (:get_tag, :get_isclosure, :get_compiled_argtypes)
+    @eval $fn(thunk::Thunk) = $fn(typeof(thunk))
+end
+
+function get_compiled_argtypes(::Type{<:Thunk{<:Any,<:Any,<:Any,ArgTypes}}) where {ArgTypes}
+    return ArgTypes
+end
+
+get_tag(::Type{<:Thunk{<:Any,tag}}) where {tag} = tag
+
+get_isclosure(::Type{<:Thunk{<:Any,<:Any,IsClosure}}) where {IsClosure} = IsClosure
+
 function Base.show(io::IO, thunk::Thunk{<:Any,tag}) where {tag}
     return print(io, "Reactant compiled function $(thunk.f) (with tag $(tag))")
 end
@@ -3752,24 +3764,13 @@ function Base.showerror(
     )
 end
 
-@generated function (
-    thunk::Thunk{FTy,tag,IsClosure,ArgTypes,ExecTy,DeviceTy,ClientTy,GD,DAM}
-)(
-    args...
-) where {FTy,tag,IsClosure,ArgTypes,ExecTy,DeviceTy,ClientTy,GD,DAM}
+@generated function (thunk::Thunk)(args...)
     FoundTypes = Tuple{args...}
-    if ArgTypes != FoundTypes
-        return quote
-            throw(
-                $(MisMatchedThunkTypeError{
-                    Thunk{FTy,tag,IsClosure,ArgTypes,ExecTy,DeviceTy,ClientTy,GD,DAM},
-                    FoundTypes,
-                }()),
-            )
-        end
+    if get_compiled_argtypes(thunk) != FoundTypes
+        return :(throw($(MisMatchedThunkTypeError{thunk,FoundTypes}())))
     end
-    body = __thunk_fwd_body_cache[tag]
-    if IsClosure
+    body = __thunk_fwd_body_cache[get_tag(thunk)]
+    if get_isclosure(thunk)
         return quote
             args = (thunk.f, args...)
             $body

src/Reactant.jl

@@ -330,6 +330,19 @@ function __init__()
         end
     end
 
+    @static if VERSION ≥ v"1.12-"
+        if ccall(:jl_generating_output, Cint, ()) == 1
+            @warn """
+            Reactant.jl currently doesn't support versions of Julia 1.12 or newer. We are
+            actively working on adding support for newer versions of Julia. For the time
+            being we recommend using 1.11 or LTS.
+
+            For latest updates, check the status of support for Julia 1.12+ at
+            https://github.com/EnzymeAD/Reactant.jl/issues/1736.
+            """ maxlog = 1
+        end
+    end
+
     return nothing
 end
 

test/basic.jl

@@ -1671,3 +1671,11 @@ init_fill_0d(x) = fill(eltype(x)(5))
     @test @jit(init_fill(x_ra)) ≈ init_fill(x)
     @test @jit(init_fill_0d(x_ra)) ≈ init_fill_0d(x)
 end
+
+@testset "Mismatched Thunk Error" begin
+    x_ra1 = Reactant.to_rarray(rand(Float32, 2))
+    x_ra2 = Reactant.to_rarray(rand(Float64, 2))
+
+    fn = @compile sum(x_ra1)
+    @test_throws Reactant.Compiler.MisMatchedThunkTypeError fn(x_ra2)
+end