15 lines
332 B
C
15 lines
332 B
C
#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}:{full_signature}:{algo_info}
|
|
CUresult{_placeholder} {kernel_name}(CUstream stream, {signature});
|