FYI/Clarification for `*_like(arr)` functions using `arr.device`

Abierto
#1,013 3 comentarios 0 reacciones 0 asignados Ver en GitHub

Nadie ha tomado este issue todavía.

Evaluación

Dificultad
5/5
Tiempo estimado
Más de una semana
Aptitud para principiantes
35/100
Tipo de issue
Nueva funcionalidad
Claridad
Necesita aclaración
Estado de actividad
Activo
Stack tecnológico
python
Área
api

Línea de trabajo

Comienza revisando el requisito de Array API para el manejo de device en *_like(arr) y el comportamiento de CuPy, JAX y torch indicado en el issue. Se considera terminado cuando los maintainers decidan si el estándar necesita una aclaración o un cambio de comportamiento, y se documente el impacto esperado en la compatibilidad.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

This may be more of a CuPy question about how comfortable CuPy is with changing behavior and pushing through the change in v15. But it does come from the Array API so just bringing it up here once for awareness.

The standard states that empty_like(arr) should allocate as if empty_like(arr, device=arr.device) was passed. And that is actually what JAX and torch do:

import cupy as cp

arr = cp.arange(1000)  # arr.device == Device(0)
with cp.cuda.Device(1)
    print(cp.empty_like(arr).device)  # Device(1) for cupy, 0 for JAX/torch

CuPy does currently not do this (I half suspect this was missed at the time), CuPy completely relies on context managers right now (to the point that even a NumPy array works there).

CC @leofang

Lenguaje dominante
Python
Estrellas
281
Forks
52
Sin PR fusionados en 30 d

Guía de contribución

Abrir la guía de contribución

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de data-apis/array-api

Todos los issues de data-apis/array-api

Issues similares

Más issues de Python

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.