cupy_nvrtc.h 419 Bytes
Newer Older
root's avatar
root committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#ifndef INCLUDE_GUARD_CUDA_CUPY_NVRTC_H
#define INCLUDE_GUARD_CUDA_CUPY_NVRTC_H

#include <cuda.h>  // for CUDA_VERSION
#include <nvrtc.h>

extern "C" {

#if CUDA_VERSION < 11010
// functions added in CUDA 11.1
nvrtcResult nvrtcGetCUBINSize(...) {
    return NVRTC_ERROR_COMPILATION;
}

nvrtcResult nvrtcGetCUBIN(...) {
    return NVRTC_ERROR_COMPILATION;
}
#endif

}

#endif // #ifndef INCLUDE_GUARD_CUDA_CUPY_NVRTC_H