JuliaGPU/CUDA.jl

Should compact `show` mark `CuArray`s?

Open

#1 190 ouverte le 6 oct. 2021

Voir sur GitHub
 (1 commentaire) (0 réactions) (0 assignés)Julia (274 forks)batch import
enhancementgood first issue

Métriques du dépôt

Stars
 (1 408 stars)
Métriques de merge PR
 (Merge moyen 5j 5h) (16 PRs mergées en 30 j)

Description

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.

Guide contributeur