| |
| |
| |
| |
| |
|
|
| from libc.stdint cimport uint32_t, uint64_t |
|
|
| cdef extern from "nvrtc.h": |
|
|
| ctypedef enum nvrtcResult: |
| NVRTC_SUCCESS = 0 |
| NVRTC_ERROR_OUT_OF_MEMORY = 1 |
| NVRTC_ERROR_PROGRAM_CREATION_FAILURE = 2 |
| NVRTC_ERROR_INVALID_INPUT = 3 |
| NVRTC_ERROR_INVALID_PROGRAM = 4 |
| NVRTC_ERROR_INVALID_OPTION = 5 |
| NVRTC_ERROR_COMPILATION = 6 |
| NVRTC_ERROR_BUILTIN_OPERATION_FAILURE = 7 |
| NVRTC_ERROR_NO_NAME_EXPRESSIONS_AFTER_COMPILATION = 8 |
| NVRTC_ERROR_NO_LOWERED_NAMES_BEFORE_COMPILATION = 9 |
| NVRTC_ERROR_NAME_EXPRESSION_NOT_VALID = 10 |
| NVRTC_ERROR_INTERNAL_ERROR = 11 |
| NVRTC_ERROR_TIME_FILE_WRITE_FAILED = 12 |
| NVRTC_ERROR_NO_PCH_CREATE_ATTEMPTED = 13 |
| NVRTC_ERROR_PCH_CREATE_HEAP_EXHAUSTED = 14 |
| NVRTC_ERROR_PCH_CREATE = 15 |
| NVRTC_ERROR_CANCELLED = 16 |
| NVRTC_ERROR_TIME_TRACE_FILE_WRITE_FAILED = 17 |
| NVRTC_ERROR_BUSY = 18 |
|
|
| cdef struct _nvrtcProgram: |
| pass |
| ctypedef _nvrtcProgram* nvrtcProgram |
|
|
| cdef struct anon_struct0: |
| int available |
| size_t compressedSize |
| size_t uncompressedSize |
| int cudaVersionMajor |
| int cudaVersionMinor |
| unsigned int numFiles |
|
|
| ctypedef anon_struct0 nvrtcBundledHeadersInfo |
|
|
| cdef const char* nvrtcGetErrorString(nvrtcResult result) except ?NULL nogil |
|
|
| cdef nvrtcResult nvrtcVersion(int* major, int* minor) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetNumSupportedArchs(int* numArchs) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetSupportedArchs(int* supportedArchs) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcCreateProgram(nvrtcProgram* prog, const char* src, const char* name, int numHeaders, const char** headers, const char** includeNames) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcDestroyProgram(nvrtcProgram* prog) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcCompileProgram(nvrtcProgram prog, int numOptions, const char** options) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetPTXSize(nvrtcProgram prog, size_t* ptxSizeRet) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetPTX(nvrtcProgram prog, char* ptx) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetCUBINSize(nvrtcProgram prog, size_t* cubinSizeRet) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetCUBIN(nvrtcProgram prog, char* cubin) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetLTOIRSize(nvrtcProgram prog, size_t* LTOIRSizeRet) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetLTOIR(nvrtcProgram prog, char* LTOIR) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetOptiXIRSize(nvrtcProgram prog, size_t* optixirSizeRet) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetOptiXIR(nvrtcProgram prog, char* optixir) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetProgramLogSize(nvrtcProgram prog, size_t* logSizeRet) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetProgramLog(nvrtcProgram prog, char* log) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcAddNameExpression(nvrtcProgram prog, const char* name_expression) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetLoweredName(nvrtcProgram prog, const char* name_expression, const char** lowered_name) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetPCHHeapSize(size_t* ret) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcSetPCHHeapSize(size_t size) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetPCHCreateStatus(nvrtcProgram prog) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetPCHHeapSizeRequired(nvrtcProgram prog, size_t* size) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcSetFlowCallback(nvrtcProgram prog, void* callback, void* payload) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetTileIRSize(nvrtcProgram prog, size_t* TileIRSizeRet) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetTileIR(nvrtcProgram prog, char* TileIR) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcInstallBundledHeaders(const char* installPath, unsigned int flags, const char** errorLog) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcGetBundledHeadersInfo(nvrtcBundledHeadersInfo* info, const char** errorLog) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef nvrtcResult nvrtcRemoveBundledHeaders(const char* installPath, const char** errorLog) except ?NVRTC_ERROR_INVALID_INPUT nogil |
|
|
| cdef enum: NVRTC_INSTALL_HEADERS_SKIP_IF_EXISTS = 0 |
|
|
| cdef enum: NVRTC_INSTALL_HEADERS_FORCE_OVERWRITE = 1 |
|
|
| cdef enum: NVRTC_INSTALL_HEADERS_NO_WAIT = 2 |
|
|