Add a make_zero path for TracedRArray

src/Reactant.jl

@@ -138,6 +138,13 @@ end
         seen[prev] = res
         return res
     end
+
+    if RT <: TracedRArray
+        @assert typeof(prev) <: RT # prev has concrete shape while RT might not have that
+        res = typeof(prev)(prev.paths, prev.mlir_data)
+        seen[prev] = res
+        return res
+    end
 
     attr = fill(MLIR.IR.Attribute(eltype(RT)(0)), mlir_type(prev))
     cst = MLIR.IR.result(MLIR.Dialects.stablehlo.constant(value=attr), 1)
@@ -146,8 +153,6 @@ end
     return res
 end
 
-
-
 function Base.promote_rule(A::Type{TracedRArray{T, Shape, N}}, B::Type{TracedRArray{S, Shape, N}}) where {T, S, Shape, N}
     TracedRArray{Base.promote_type(T, S), Shape, N}
 end