fix: try removing nospecialize

Author: avik-pal

src/Reactant.jl

@@ -178,39 +178,37 @@ include("TracedRArray.jl")
 
 include("ConcreteRArray.jl")
 
-Base.@nospecializeinfer use_overlayed_version(x) = false
-Base.@nospecializeinfer function use_overlayed_version(
-    @nospecialize(x::F)
-) where {F<:Function}
+use_overlayed_version(x) = false
+function use_overlayed_version(x::F) where {F<:Function}
     return Base.inferencebarrier(any)(
         use_overlayed_version, getfield.(Ref(x), fieldnames(F))
     )
 end
-Base.@nospecializeinfer function use_overlayed_version(@nospecialize(x::Base.Generator))
+function use_overlayed_version(x::Base.Generator)
     return use_overlayed_version(x.f) || use_overlayed_version(x.iter)
 end
-Base.@nospecializeinfer function use_overlayed_version(@nospecialize(x::Base.Iterators.Zip))
+function use_overlayed_version(x::Base.Iterators.Zip)
     return Base.inferencebarrier(any)(use_overlayed_version, x.is)
 end
-Base.@nospecializeinfer use_overlayed_version(@nospecialize(x::Base.Iterators.Enumerate)) =
-    use_overlayed_version(x.itr)
-Base.@nospecializeinfer use_overlayed_version(@nospecialize(x::Vector)) =
-    Base.inferencebarrier(any)(use_overlayed_version, x)
-Base.@nospecializeinfer use_overlayed_version(@nospecialize(iter::Tuple)) =
-    Base.inferencebarrier(any)(use_overlayed_version, iter)
-Base.@nospecializeinfer function use_overlayed_version(@nospecialize(iter::NamedTuple))
+function use_overlayed_version(x::Base.Iterators.Enumerate)
+    return use_overlayed_version(x.itr)
+end
+function use_overlayed_version(x::Vector)
+    return Base.inferencebarrier(any)(use_overlayed_version, x)
+end
+function use_overlayed_version(iter::Tuple)
+    return Base.inferencebarrier(any)(use_overlayed_version, iter)
+end
+function use_overlayed_version(iter::NamedTuple)
     return Base.inferencebarrier(any)(use_overlayed_version, values(iter))
 end
-Base.@nospecializeinfer use_overlayed_version(@nospecialize(x::TracedRArray)) = true
-Base.@nospecializeinfer use_overlayed_version(@nospecialize(x::TracedRNumber)) = true
-Base.@nospecializeinfer use_overlayed_version(@nospecialize(x::Number)) = false
-Base.@nospecializeinfer use_overlayed_version(@nospecialize(x::MissingTracedValue)) = true
-Base.@nospecializeinfer use_overlayed_version(
-    @nospecialize(x::AbstractArray{<:TracedRNumber})
-) = true
-Base.@nospecializeinfer use_overlayed_version(@nospecialize(rng::ReactantRNG)) =
-    use_overlayed_version(rng.seed)
-Base.@nospecializeinfer function use_overlayed_version(@nospecialize(x::AbstractArray))
+use_overlayed_version(x::TracedRArray) = true
+use_overlayed_version(x::TracedRNumber) = true
+use_overlayed_version(x::Number) = false
+use_overlayed_version(x::MissingTracedValue) = true
+use_overlayed_version(x::AbstractArray{<:TracedRNumber}) = true
+use_overlayed_version(rng::ReactantRNG) = use_overlayed_version(rng.seed)
+function use_overlayed_version(x::AbstractArray)
     a = ancestor(x)
     a === x && return false
     return use_overlayed_version(a)