Triton JIT support for device APIs
未关闭
还没有人认领这个 Issue。
- 主要语言
- Cython
- 星标
- 601
- 派生
- 46
- PR 合并指标
- 30 天内没有已合并 PR
描述
There is already support for Numba CUDA JIT but Triton JIT interacts much better with frameworks like PyTorch.
贡献指南
从这里开始
- 先读完整个 Issue,再读项目的贡献指南。
- 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
- Fork 仓库,在一个分支上完成修改。
- 提交 Pull Request,并在描述里引用这个 Issue 编号。
调研方向
该 issue 未指定文件、测试或入口点。首先定位现有的 Numba CUDA JIT 支持,并检查 device API 的暴露方式,然后将 Triton JIT 的要求与 PyTorch 进行比较。完成的标准是:相关 device API 已提供 Triton JIT 支持,并且覆盖受支持的使用场景。
由索引模型根据 Issue 内容生成。
评估
- 技术栈
- python, pytorch
- 领域
- backend, machine-learning
- Issue 类型
- 功能
- 难度
- 5/5
- 预计耗时
- 一周以上
- 活跃度
- 停滞
- 描述清晰度
- 需要澄清
- 新手友好度
- 25/100