Ejemplos

Si no estás seguro de por dónde empezar, o nunca has visto un archivo PTX, quizás quieras revisar algunos kernels de ejemplo en el directorio examples/. Estos incluyen:

• attention.ptx — kernel del mecanismo de atención
• elu.ptx — función de activación ELU (Exponential Linear Unit)
• gele.ptx — función de activación GELU (Gaussian Error Linear Unit)
• gemm.ptx — Multiplicación General de Matrices
• histogram.ptx — kernel de cálculo de histograma
• layernorm.ptx — kernel de Layer Normalization
• matrix.ptx — operaciones con matrices
• multi_tensor.ptx — operaciones multi-tensor
• reduce_sum.ptx — kernel de suma por reducción paralela
• relu.ptx — función de activación ReLU
• softmax.ptx — función de activación Softmax

Limitaciones

PTX está basado en un lenguaje de formato de texto y la salida de nvdisasm/cuobjdump ya contiene bastante información que se inserta en los binarios. Trabajar con texto requiere construir una gramática precisa para probar el parser, sin embargo existen algunos límites:

• las etiquetas ocasionalmente no coinciden con su instrucción objetivo, lo que puede dar lugar a una visualización incorrecta del flujo de control en Binary Ninja
• las declaraciones globales declaradas fuera del ámbito de una función no están soportadas
• el espacio de memoria asignado fuera de las funciones no está mapeado
• no todas las operaciones atómicas (atom) están totalmente soportadas
• los nombres de los símbolos no están desenmascarados — la embellecimiento de cuobjdump/cufilt aún no se aplica

Contacto

Para más información, contacta con Nicolò Altamura (@nicolodev).