Projetando Núcleos de GPU de Alto Desempenho com TileLang: Tensor GEMMs, Softmax Fundido, FlashAttention e Ajuste Automático
TileLang é uma linguagem específica de domínio de alto nível em Python para simplificar a criação de núcleos de GPU. O guia demonstra a implementação passo a passo de tarefas complexas: tensor GEMMs, softmax fundido e FlashAttention, onde o compilador lida com detalhes de baixo nível do CUDA.
MarkTechPost26.07 · 15:02



