cupy_cuda_runtime.h 1.73 KB
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
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#ifndef INCLUDE_GUARD_CUDA_CUPY_CUDA_RUNTIME_H
#define INCLUDE_GUARD_CUDA_CUPY_CUDA_RUNTIME_H

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

extern "C" {

bool hip_environment = false;

#if CUDA_VERSION < 10010
const int cudaErrorContextIsDestroyed = 709;
#endif

#if CUDA_VERSION < 11010
// APIs added in CUDA 11.1
cudaError_t cudaGraphUpload(...) {
    return cudaErrorUnknown;
}
#endif

#if CUDA_VERSION < 11020
// APIs added in CUDA 11.2
typedef void* cudaMemPool_t;
enum cudaMemPoolAttr {};
enum cudaMemAllocationType {};
enum cudaMemAllocationHandleType {};
enum cudaMemLocationType {};
struct cudaMemLocation {
    int id;
    cudaMemLocationType type;
};
struct cudaMemPoolProps {
    cudaMemAllocationType allocType;
    cudaMemAllocationHandleType handleTypes;
    struct cudaMemLocation location;
    unsigned char reserved[64];
    void* win32SecurityAttributes;
};

cudaError_t cudaMallocAsync(...) {
    return cudaErrorUnknown;
}

cudaError_t cudaMallocFromPoolAsync(...) {
    return cudaErrorUnknown;
}

cudaError_t cudaFreeAsync(...) {
    return cudaErrorUnknown;
}

cudaError_t cudaDeviceSetMemPool(...) {
    return cudaErrorUnknown;
}

cudaError_t cudaDeviceGetDefaultMemPool(...) {
    return cudaErrorUnknown;
}

cudaError_t cudaDeviceGetMemPool(...) {
    return cudaErrorUnknown;
}

cudaError_t cudaMemPoolCreate(...) {
    return cudaErrorUnknown;
}

cudaError_t cudaMemPoolDestroy(...) {
    return cudaErrorUnknown;
}

cudaError_t cudaMemPoolTrimTo(...) {
    return cudaErrorUnknown;
}

cudaError_t cudaMemPoolSetAttribute(...) {
    return cudaErrorUnknown;
}

cudaError_t cudaMemPoolGetAttribute(...) {
    return cudaErrorUnknown;
}

#endif

} // extern "C"

#endif // #ifndef INCLUDE_GUARD_CUDA_CUPY_CUDA_RUNTIME_H