Update ext/ReactantCUDAExt.jl

Co-authored-by: jumerckx <[email protected]>
EnzymeAD · Jan 8, 2025 · 839050b · 839050b
1 parent 5fd4c39
commit 839050b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ext/ReactantCUDAExt.jl b/ext/ReactantCUDAExt.jl
@@ -363,7 +363,7 @@ function to_bytes(x)
 	end
 end
 
-function Reactant.make_tracer(seen, @nospecialize(prev::CuTracedArray{T,N}), @nospecialize(path), mode; kwargs...)
+function Reactant.make_tracer(seen, @nospecialize(prev::CuTracedArray), @nospecialize(path), mode; kwargs...)
     x = Base.unsafe_pointer_to_objref(Base.reinterpret(Ptr{Cvoid}, prev.ptr))::TracedRArray
     return Reactant.make_tracer(seen, x, path, mode; kwargs...)
 end