Fix dispatch match on IO type of show on TracedRArray

EnzymeAD · Jul 21, 2024 · 405c7fd · 405c7fd
1 parent 1fbd920
commit 405c7fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Tracing.jl b/src/Tracing.jl
@@ -24,7 +24,7 @@ function Base.similar(x::TracedRArray{T,N}, ::Type{T2}) where {T,N,T2}
     return TracedRArray{T2,N}((), nothing, size(x))
 end
 
-function Base.show(io::IO, X::TracedRArray{T,N}) where {T,N}
+function Base.show(io::IOty, X::TracedRArray{T,N}) where {T,N,IOty<:Union{IO,IOContext}}
     print(io, "TracedRArray{", T, ",", N, "N}(", X.paths, ", ")
     return print(io, X.mlir_data, ")")
 end