Hacktoberfest 2026: le issue che i maintainer hanno segnato per ottobre, aperte e adatte ai principianti. Sfoglia le issue Hacktoberfest

Fused gemm + comm for CP A2A on Blackwell

Aperta
  • Qualcosa non va?
#3,664 0 commenti 0 reazioni 1 assegnatario Vedi su GitHub

I maintainer di solito rispondono entro 2 giorni

@cyanguwa ci sta già lavorando.

Dal 9/10/2026.

Valutazione

Questa issue non è ancora stata valutata.

Descrizione

2.22 attention

To make CP A2A more efficient, we can fuse gemms and the communication.

Affected module:

  • MHA: QKV Linear + DPA (CP/non-CP) + Proj Linear
    Right now:
  • QKV Linear produces (B,S/cp_rank,H,D), DPA CP gathers S and splits along H to get (B,S,H/cp_rank,D), before the actual attention kernel, ...
    After fusion:
  • QKV Linear produces (B,S,H/cp_rank,D), DPA CP just needs to check if the right shards have arrived, before calling the kernel, ...

We have some Blackwell kernels ready, and will need to look into CP no_load_balancing strategy and integrate those kernels.

Lingua principale
Python
Stelle
3.6k
Fork
851
Merge medio
4g 20h
PR unite (30g)
58

Preparare l'ambiente

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Altre issue di NVIDIA/TransformerEngine

Tutte le issue di NVIDIA/TransformerEngine

Issue simili

Altre issue su Python

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.