Exemplos

Se você não tem certeza de por onde começar, ou nunca viu um arquivo PTX, talvez queira conferir alguns kernels de exemplo no diretório examples/. Estes incluem:

• attention.ptx — kernel do mecanismo de atenção
• elu.ptx — função de ativação ELU (Exponential Linear Unit)
• gele.ptx — função de ativação GELU (Gaussian Error Linear Unit)
• gemm.ptx — Multiplicação Geral de Matrizes
• histogram.ptx — kernel de cálculo de histograma
• layernorm.ptx — kernel de Normalização de Camada
• matrix.ptx — operações com matrizes
• multi_tensor.ptx — operações com múltiplos tensores
• reduce_sum.ptx — kernel de soma por redução paralela
• relu.ptx — função de ativação ReLU
• softmax.ptx — função de ativação Softmax

Limitações

O PTX é baseado em uma linguagem de formato de texto e a saída do nvdisasm/cuobjdump já possui muitas informações que são inseridas nos binários. Lidar com texto exige a construção de uma gramática precisa para testar o parser, no entanto existem algumas limitações:

• rótulos ocasionalmente são incompatíveis com sua instrução de destino, o que pode resultar em visualização incorreta do fluxo de controle no Binary Ninja
• declarações globais declaradas fora do escopo da função não são suportadas
• memória alocada fora de funções não é mapeada
• nem todas as operações atômicas (atom) são totalmente suportadas
• nomes de símbolos não são desmembrados — a beautificação do cuobjdump/cufilt ainda não é aplicada

Contato

Para mais informações, entre em contato com Nicolò Altamura (@nicolodev).