compile.h 332 B

1234567891011121314
  1. #ifndef TT_KERNEL_INCLUDES
  2. #define TT_KERNEL_INCLUDES
  3. #include <cuda.h>
  4. #include <inttypes.h>
  5. #include <stdint.h>
  6. #include <stdio.h>
  7. #endif
  8. void unload_{kernel_name}(void);
  9. void load_{kernel_name}(void);
  10. // tt-linker: {kernel_name}:{full_signature}:{algo_info}
  11. CUresult{_placeholder} {kernel_name}(CUstream stream, {signature});