Hacktoberfest 2026: los issues que los mantenedores marcaron para octubre, abiertos y aptos para principiantes. Explorar issues de Hacktoberfest

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

Abierto
#577 0 comentarios 0 reacciones 0 asignados Ver en GitHub

Los mantenedores suelen responder en 3 días

Nadie ha tomado este issue todavía.

Evaluación

Dificultad
5/5
Tiempo estimado
Más de una semana
Aptitud para principiantes
15/100
Tipo de issue
Nueva funcionalidad
Claridad
Necesita aclaración
Estado de actividad
Activo
Stack tecnológico
python, pytorch

Línea de trabajo

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.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

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

Lenguaje dominante
Python
Estrellas
202
Forks
52
Merge medio
5 d 2 h
PR fusionados (30 d)
12

Preparar el entorno

Abrir en Codespaces

Inicia el contenedor de desarrollo del proyecto en tu navegador, con tu propia cuenta de GitHub.

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de ROCm/iris

Todos los issues de ROCm/iris

Issues similares

Más issues de Python

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.