device(::Type{<:CuArrays.CuArray}) = GPU

chriselrod · chriselrod · commit a714df91f6f4 · 2021-06-03T09:30:13.000-04:00
diff --git a/src/cuarrays.jl b/src/cuarrays.jl
@@ -14,5 +14,5 @@ function restructure(x::CuArrays.CuArray, y)
     return reshape(Adapt.adapt(parameterless_type(x), y), Base.size(x)...)
 end
 
-Device(::Type{<:CuArrays.CuArray}) = GPU()
+device(::Type{<:CuArrays.CuArray}) = GPU()
 
