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

[Feature]: follow-up to PyTorch allocation provider to accept already rendezvoused tensors

Aperta
#577 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub

I maintainer di solito rispondono entro 2 giorni

Nessuno ha ancora preso questa issue.

Valutazione

Difficoltà
5/5
Tempo stimato
Più di una settimana
Idoneità per principianti
15/100
Tipo di issue
Funzionalità
Chiarezza
Da chiarire
Stato di attività
Attiva
Stack tecnologico
python, pytorch

Direzione di ricerca

The payload names no source files or tests. It refers to the earlier PyTorch allocation provider in #551, torch.distributed._symmetric_memory, and the single-map iris.copy limitation, so start by reading #551 and the provider it added. Done would need an agreed API for accepting pre-rendezvoused tensors, which the issue does not yet define, so settle that design with maintainers first.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Descrizione

iris
Suggestion Description

#551 provides capability to run Iris device kernels with memory allocated via PyTorch, with no copy, no second heap, and no kernel changes.
and we agreed on extending it to accept tensors frameworks have already rendezvoused , because users might want to manage the memory.

Overall Motivation behind #551 and this feature request is as follows:

  • Some of the inference stack already uses torch.distributed.symmetric_memory. TokenSpeed's amd_rsag* path, the TritonAllReduceBackend, and the torch.ops.symm_mem.* collectives that vLLM and SGLang-style engines may call all allocate through it. Until now, using an Iris kernel meant moving those buffers into the Iris heap. This provider lets the two coexist: a framework keeps its torch allocations, and individual Iris kernels can run on them.
  • Frameworks that already own their buffers can't use Iris device kernels until the follow-up lands.
  • The single-map iris.copy limitation also means it does not yet make every Iris operation work across independently allocated tensors
Operating System

No response

GPU

No response

ROCm Component

No response

Lingua principale
Python
Stelle
202
Fork
52
Merge medio
5g 2h
PR unite (30g)
23

Preparare l'ambiente

Apri in Codespaces

Avvia il container di sviluppo del progetto nel browser, con il tuo account GitHub.

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 ROCm/iris

Tutte le issue di ROCm/iris

Issue simili

Altre issue su Python

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.