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

未关闭
#1,013 3 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看

还没有人认领这个 Issue。

评估

难度
5/5
预计耗时
一周以上
新手友好度
35/100
Issue 类型
功能
描述清晰度
需要澄清
活跃度
活跃
技术栈
python
领域
api

调研方向

首先，审查 Array API 对 *_like(arr) 的 device 处理要求，以及 issue 中报告的 CuPy、JAX 和 torch 行为。完成的标准是维护者决定该标准是否需要澄清或行为变更，并记录预期的兼容性影响。

由索引模型根据 Issue 内容生成。

描述

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

主要语言
Python
星标
281
派生
52
30 天内没有已合并 PR

贡献指南

打开贡献指南

从这里开始

  1. 先读完整个 Issue，再读项目的贡献指南。
  2. 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
  3. Fork 仓库，在一个分支上完成修改。
  4. 提交 Pull Request，并在描述里引用这个 Issue 编号。

data-apis/array-api 的其他 Issue

查看 data-apis/array-api 的全部 Issue

相似的 Issue

更多 Python Issue

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。