Esempi

Se non sei sicuro da dove iniziare, o non hai mai visto un file PTX, potresti voler dare un'occhiata ad alcuni kernel di esempio nella directory examples/. Questi includono:

• attention.ptx — kernel del meccanismo di attenzione
• elu.ptx — funzione di attivazione ELU (Exponential Linear Unit)
• gele.ptx — funzione di attivazione GELU (Gaussian Error Linear Unit)
• gemm.ptx — Moltiplicazione Generale di Matrici
• histogram.ptx — kernel di calcolo dell'istogramma
• layernorm.ptx — kernel di Layer Normalization
• matrix.ptx — operazioni su matrici
• multi_tensor.ptx — operazioni multi-tensore
• reduce_sum.ptx — kernel di somma con riduzione parallela
• relu.ptx — funzione di attivazione ReLU
• softmax.ptx — funzione di attivazione Softmax

Limitazioni

PTX è basato su un linguaggio in formato testuale e l'output di nvdisasm/cuobjdump contiene già molte informazioni che vengono inserite nei binari. Gestire il testo richiede la costruzione di una grammatica precisa per testare il parser, tuttavia ci sono alcuni limiti:

• le etichette occasionalmente non corrispondono alla loro istruzione target, il che può comportare una visualizzazione errata del flusso di controllo in Binary Ninja
• le istruzioni globali dichiarate al di fuori dello scope della funzione non sono supportate
• la memoria allocata al di fuori delle funzioni non viene mappata
• non tutte le operazioni atomiche (atom) sono completamente supportate
• i nomi dei simboli non vengono demangled — la beautificazione di cuobjdump/cufilt non è ancora applicata

Contatti

Per ulteriori informazioni, contatta Nicolò Altamura (@nicolodev).