Ref ptr fix (#698)

* Ref ptr fix

* remove eval

* Fix comma

* syntax fix

* fix

* fix

* Apply formatting suggestions from code review

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

---------

Co-authored-by: Mosè Giordano <765740+giordano@users.noreply.github.com>
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

Author: 3 people

src/Tracing.jl

@@ -366,15 +366,41 @@ end
 
 for P in (Ptr, Core.LLVMPtr, Base.RefValue)
     @eval Base.@nospecializeinfer function traced_type_inner(
-        @nospecialize(PT::Type{<:$P}),
+        @nospecialize(PT::Type{$P}),
         seen,
         mode::TraceMode,
         @nospecialize(track_numbers::Type)
     )
-        T = eltype(PT)
-        return $P{traced_type_inner(T, seen, mode, track_numbers)}
+        return $P
     end
 end
+for P in (Ptr, Base.RefValue)
+    @eval Base.@nospecializeinfer function traced_type_inner(
+        @nospecialize(PT::Type{$P{T}}),
+        seen,
+        mode::TraceMode,
+        @nospecialize(track_numbers::Type)
+    ) where {T}
+        return $P{traced_type_inner(PT.parameters[1], seen, mode, track_numbers)}
+    end
+end
+
+Base.@nospecializeinfer function traced_type_inner(
+    @nospecialize(PT::Type{Core.LLVMPtr{T}}),
+    seen,
+    mode::TraceMode,
+    @nospecialize(track_numbers::Type)
+) where {T}
+    return Core.LLVMPtr{traced_type_inner(PT.body.parameters[1], seen, mode, track_numbers)}
+end
+Base.@nospecializeinfer function traced_type_inner(
+    @nospecialize(PT::Type{Core.LLVMPtr{T,A}}),
+    seen,
+    mode::TraceMode,
+    @nospecialize(track_numbers::Type)
+) where {T,A}
+    return Core.LLVMPtr{traced_type_inner(PT.parameters[1], seen, mode, track_numbers),A}
+end
 
 Base.@nospecializeinfer function traced_type_inner(
     @nospecialize(VT::Type{<:Val}),

test/tracing.jl

@@ -165,6 +165,11 @@ end
                     NTuple{nsteps,SpectralVariable3D} where {nsteps,SpectralVariable3D},
                     NTuple{nsteps,SpectralVariable3D} where {nsteps,SpectralVariable3D},
                 ),
+                (
+                    Base.RefValue{A} where {A},
+                    Base.RefValue{A} where {A},
+                    Base.RefValue{A} where {A},
+                ),
                 (Wrapper{Symbol,Symbol}, Wrapper{Symbol,Symbol}, Wrapper{Symbol,Symbol}),
                 (
                     Wrapper{Float64,Vector{Float64}},