Julia JIT support for device APIs
まだ誰も着手していません。
- 主要言語
- 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 にコメントします — 二人が同じ作業をするのを防げます。
- リポジトリをフォークし、ブランチを切って変更します。
- issue 番号を参照したプルリクエストを送ります。
調査の方向性
まず、リンクされている cuBLASDx DGEMM emulation sample と関連する Julia の議論を読んでください。この issue では、リポジトリのファイル、テスト、具体的なエントリポイントが挙げられておらず、libmathdx、PTX、または NVVM IR を Julia カーネルにリンクするための受け入れテストも定義されていません。
索引モデルが issue の本文から書いたものです。
評価
- 技術スタック
- cpp, julia, python
- 領域
- compilers, hpc
- issue の種類
- 機能追加
- 難易度
- 5/5
- 見積もり時間
- 1週間以上
- 活発さ
- 停滞
- 明瞭さ
- 説明が足りない
- 初心者へのやさしさ
- 25/100