[Feature] Support encoder models for flash_infer backend

Métricas do repositório

Stars

28.442 estrelas (28.442 estrelas)

Métricas de merge de PR

Mesclagem média 1d 13h (Mesclagem média 1d 13h)1.000 fundiu PRs em 30d (1.000 fundiu PRs em 30d)

Description

          @DavidBao03 hi, currently only support encoder model with torch_native attn backend and triton attn backend. Other attn backend is not supported yet.

Guia do colaborador

Direção de pesquisa

Investigue como os modelos de codificador são suportados nos backends torch native e triton, e então estenda o backend flash infer de forma similar.

Pilha de tecnologia

pythonpytorch

Domain

backend

Tipo Issue

Funcionalidade

DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.

3

Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.

3-5 dias

Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.

Ativo

ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.

Claro

Prerequisites

PythonPyTorch

Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.

50