Viewing File: /home/ubuntu/codegamaai-test/rag_drive/regdrive/lib/python3.10/site-packages/triton/tools/compile.h

#ifndef TT_KERNEL_INCLUDES
#define TT_KERNEL_INCLUDES

#include <cuda.h>
#include <inttypes.h>
#include <stdint.h>
#include <stdio.h>

#endif

void unload_{kernel_name}(void);
void load_{kernel_name}(void);
// tt-linker: {kernel_name}:{signature}
CUresult{_placeholder} {kernel_name}(CUstream stream, unsigned int gX,
                                     unsigned int gY, unsigned int gZ,
                                     {signature});