feat: use traced_call when unrolling iterators and generators (#1642)

* feat: better support for Base.Generators

* feat: use traced_call when unrolling iterators and generators

* fix: closure with call working

* fix: try removing nospecialize

* fix: use a looped version of any to avoid inference issues

* fix: dont overlay inside compiler call

Author: avik-pal

src/Reactant.jl

@@ -193,7 +193,6 @@ use_overlayed_version(::TracedRArray) = true
 use_overlayed_version(::TracedRNumber) = true
 use_overlayed_version(::Number) = false
 use_overlayed_version(::MissingTracedValue) = true
-use_overlayed_version(::Vector{<:AnyTracedRArray}) = true
 use_overlayed_version(::AbstractArray{<:TracedRNumber}) = true
 use_overlayed_version(rng::ReactantRNG) = use_overlayed_version(rng.seed)
 function use_overlayed_version(x::AbstractArray)

src/TracedRArray.jl

@@ -1532,7 +1532,6 @@ end
 
 unwrapped_broadcast(f::F, xs) where {F} = unrolled_map(f, xs)
 
-# TODO: once traced_call supports internal mutations, we can use traced_call here
 # TODO: we should overload this for Slices and use mapslices instead
 function unrolled_map(f::F, itr) where {F}
     y = Reactant.call_with_reactant(iterate, itr)