mirage-project/mirage

[Feature Request] - Optimizing Shared Memory Usage in multitoken_paged_attention

Open

#450 aberto em 11 de ago. de 2025

Ver no GitHub
 (1 comment) (0 reactions) (0 assignees)Cuda (232 forks)auto 404
good first issue

Métricas do repositório

Stars
 (2.381 stars)
Métricas de merge de PR
 (Métricas PR pendentes)

Description

Feature Description

Our multitoken_paged_attention task cannot process large number of tokens. We should reduce shared memory usage in this task, which would allow us to run prefill using larger batch sizes.

Guia do colaborador