CUDA.jl
CUDA.jl copied to clipboard
Should compact `show` mark `CuArray`s?
trafficstars
The compact printing used within a Tuple etc. doesn't at present distinguish a CuArray from an ordinary Array:
julia> (; x = cu([1f0])) # CUDA v3.4.2
(x = Float32[1.0],)
julia> (; x = [1f0])
(x = Float32[1.0],)
Maybe it should? This came up in https://github.com/FluxML/Zygote.jl/issues/1095 , but more generally Zygote likes to return tuples of arrays, and having a visible distinction might be helpful.
This could just be cu(Float32[1.0]) but perhaps that's not right for all eltypes.
Ha, adding cu is clever. It might not be entirely correct, but for compact printing it's strictly better than the confusing status quo.