Proper traced type inner of subarray (#946)

* Proper traced type inner of subarray

* no bool

* Apply suggestions from code review

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

---------

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

Author: wsmoses

src/Tracing.jl

@@ -551,6 +551,20 @@ Base.@nospecializeinfer function traced_type_inner(
     end
 end
 
+Base.@nospecializeinfer function Reactant.traced_type_inner(
+    @nospecialize(OA::Type{SubArray{T,N,P,I,L}}),
+    seen,
+    mode::Reactant.TraceMode,
+    @nospecialize(track_numbers::Type),
+    @nospecialize(sharding),
+    @nospecialize(runtime)
+) where {T,N,P,I,L}
+    P2 = Reactant.traced_type_inner(P, seen, mode, track_numbers, sharding, runtime)
+    I2 = Reactant.traced_type_inner(I, seen, mode, track_numbers, sharding, runtime)
+    T2 = eltype(P2)
+    return SubArray{T2,N,P2,I2,L}
+end
+
 for P in (Ptr, Core.LLVMPtr, Base.RefValue)
     @eval Base.@nospecializeinfer function traced_type_inner(
         @nospecialize(PT::Type{$P}),