Fix reshape.

JuliaGPU · Sep 11, 2023 · 7e37a56 · 7e37a56
1 parent 9b1d55d
commit 7e37a56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/device/array.jl b/src/device/array.jl
@@ -260,6 +260,6 @@ end
 
 # create a derived device array (reinterpreted or reshaped) that's still a CuDeviceArray
 @inline function _derived_array(a::CuDeviceArray{<:Any,<:Any,A,I}, ::Type{T},
-                                osize::Dims{N}) where {T, N, A}
+                                osize::Dims{N}) where {T, N, A, I}
   return CuDeviceArray{T,N,A,I}(a.ptr, osize, a.maxsize)
 end