BUG: `setdiff1d` for jax.jit and dask NaN-shaped arrays
维护者通常 1 天内回复
评估
调研方向
定位 setdiff1d 的实现及其 xp.unique* 调用，然后首先在 jax.jit 中复现该失败。按照描述，使用 NaN 形状的 x1 和 x2 检查 Dask 后端，并为该情况添加单元测试。完成的标准是 setdiff1d 在报告的两种后端场景下都能正常工作，并且该回归问题已被测试覆盖。
由索引模型根据 Issue 内容生成。
描述
Follow-up to https://github.com/data-apis/array-api-extra/pull/115
setdiff1dfails when executed inside jax.jit. This is because of a call toxp.unique*without an explicit size= parameter.- By reading the code I'm fairly sure that the function fails if x1 and/or x2 have Dask backend and NaN in their shapes. Lacks a unit test.
- 主要语言
- Python
- 星标
- 33
- 派生
- 33
- 平均合并
- 1 小时 1 分钟
- 30 天内合并 PR
- 37
环境准备
- 没有 Dockerfile 或 Docker Compose 文件
- 有 Pull Request 模板
- 没有贡献指南
从这里开始
- 先读完整个 Issue，再读项目的贡献指南。
- 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
- Fork 仓库，在一个分支上完成修改。
- 提交 Pull Request，并在描述里引用这个 Issue 编号。
data-apis/array-api-extra 的其他 Issue
-
难度 2/5 1-3 小时 新手友好度 76/100
data-apis/array-api-extra#1005 · 15 条评论 ·
维护者通常 1 天内回复
-
enhancement new function
难度 2/5 1-3 小时 新手友好度 72/100
data-apis/array-api-extra#992 · 1 条评论 ·
维护者通常 1 天内回复
-
ENH, TST: Add support for MLX as array backend可能已有人在做 @prady0t 于 2 天前认领。 未关闭
难度 4/5 3-5 天 新手友好度 35/100
data-apis/array-api-extra#1006 ·
维护者通常 1 天内回复
-
难度 5/5 一周以上 新手友好度 35/100
data-apis/array-api-extra#1002 ·
维护者通常 1 天内回复
-
testing
难度 3/5 1-2 天 新手友好度 55/100
data-apis/array-api-extra#901 · 3 条评论 ·
维护者通常 1 天内回复
查看 data-apis/array-api-extra 的全部 Issue
相似的 Issue
-
难度 1/5 1 小时以内 新手友好度 85/100
-
难度 1/5 1 小时以内 新手友好度 75/100
-
难度 1/5 1 小时以内 新手友好度 85/100
-
难度 1/5 1 小时以内 新手友好度 85/100
data-umbrella/du-event-board#225 ·
-
难度 2/5 1-3 小时 新手友好度 75/100