Julia JIT support for device APIs
还没有人认领这个 Issue。
- 主要语言
- Cython
- 星标
- 601
- 派生
- 46
- PR 合并指标
- 30 天内没有已合并 PR
描述
As header-only libraries based on CUTLASS, these only work in C++.
It would be nice to have e.g. the new DGEMM via IMMA (https://github.com/NVIDIA/CUDALibrarySamples/tree/master/MathDx/cuBLASDx/16_dgemm_emulation) inside Julia kernels.
Unfortunately, Warp/numba can only JIT compile a subset of python types, and not the (more general) Julia types that we require (e.g. differential equation solvers).
A potential way forward would be to compile libmathdx to PTX, link it in via LLVM, and get full performance via LTO. However, these often contain NVVM IR, which LLVM cannot handle.
Related discussion: https://discourse.julialang.org/t/using-cublasdx-in-julia/125527
贡献指南
从这里开始
- 先读完整个 Issue，再读项目的贡献指南。
- 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
- Fork 仓库，在一个分支上完成修改。
- 提交 Pull Request，并在描述里引用这个 Issue 编号。
调研方向
首先阅读链接的 cuBLASDx DGEMM emulation sample 以及相关的 Julia 讨论。该 issue 没有指定任何 repository 文件、测试或具体的入口点，也没有定义将 libmathdx、PTX 或 NVVM IR 链接到 Julia 内核中的验收测试。
由索引模型根据 Issue 内容生成。
评估
- 技术栈
- cpp, julia, python
- 领域
- compilers, hpc
- Issue 类型
- 功能
- 难度
- 5/5
- 预计耗时
- 一周以上
- 活跃度
- 停滞
- 描述清晰度
- 需要澄清
- 新手友好度
- 25/100