Next fix for gemm-allreduce two-shot · flashinfer-ai/flashinfer#2359 | Good First Issue

(10 comments) (0 reactions) (1 assignee)Python (1,031 forks)github user discovery

cute-dslgood first issueop: commop: gemmrefactor

Repository metrics

Stars: (5,756 stars)
PR merge metrics: (平均マージ 11d 23h) (30d で 186 merged PRs)

説明

Previously: https://github.com/flashinfer-ai/flashinfer/pull/2171 https://github.com/NVIDIA/cutlass/issues/2845

Next: Update according to https://github.com/NVIDIA/cutlass/blob/main/examples/python/CuTeDSL/distributed/distributed_gemm_all_reduce_blackwell.py

コントリビューターガイド

調査方針: CUTLASSの分散gemm all reduceの例を調査し、以前のPR #2171およびCUTLASS issue #2845のパターンに従って、flashinferコードベースに適応させてください。
技術スタック: pythoncpp
領域: aibackendperformance
Issue 種別: バグ
難度: 4
推定時間: 1-2日
活動状況: アクティブ
明確さ: おおむね明確
前提条件: CUDAC++Python
初心者向け度: 15