flashinfer-ai/flashinfer

Next fix for gemm-allreduce two-shot

Open

#2359 aperta il 15 gen 2026

Vedi su GitHub
 (10 commenti) (0 reazioni) (1 assegnatario)Python (1031 fork)github user discovery
cute-dslgood first issueop: commop: gemmrefactor

Metriche repository

Star
 (5756 star)
Metriche merge PR
 (Merge medio 11g 23h) (186 PR mergiate in 30 g)

Descrizione

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

Guida contributor

Direzione di ricerca
Studia l'esempio di gemm all reduce distribuito di CUTLASS e adattalo al codebase di flashinfer, seguendo i pattern delle PR precedenti #2171 e dell'issue CUTLASS #2845.
Tech stack
pythoncpp
Dominio
aibackendperformance
Tipo issue
Bug
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
4
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-2 giorni
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Abbastanza chiara
Prerequisiti
CUDAC++Python
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
15