flashinfer-ai/flashinfer

Next fix for gemm-allreduce two-shot

Open

#2 359 ouverte le 15 janv. 2026

Voir sur GitHub
 (10 commentaires) (0 réactions) (1 assigné)Python (1 031 forks)github user discovery
cute-dslgood first issueop: commop: gemmrefactor

Métriques du dépôt

Stars
 (5 756 stars)
Métriques de merge PR
 (Merge moyen 11j 23h) (186 PRs mergées en 30 j)

Description

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

Guide contributeur

Direction de recherche
Étudiez l'exemple CUTLASS de gemm all reduce distribué et adaptez le à la base de code flashinfer, en suivant les modèles des PRs précédentes #2171 et de l'issue CUTLASS #2845.
Stack technique
pythoncpp
Domaine
aibackendperformance
Type d'issue
Bug
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
4
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-2 jours
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Plutôt claire
Prérequis
CUDAC++Python
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
15