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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
|
#!/usr/bin/env python
"""
Python interface to CUDA driver functions.
"""
import sys, ctypes
# Load library:
if 'linux' in sys.platform:
_libcuda_libname_list = ['libcuda.so']
elif sys.platform == 'darwin':
_libcuda_libname_list = ['libcuda.dylib']
elif sys.platform == 'win32':
_libcuda_libname_list = ['cuda.dll', 'nvcuda.dll']
else:
raise RuntimeError('unsupported platform')
# Print understandable error message when library cannot be found:
_libcuda = None
for _libcuda_libname in _libcuda_libname_list:
try:
if sys.platform == 'win32':
_libcuda = ctypes.windll.LoadLibrary(_libcuda_libname)
else:
_libcuda = ctypes.cdll.LoadLibrary(_libcuda_libname)
except OSError:
pass
else:
break
if _libcuda == None:
raise OSError('CUDA driver library not found')
# Exceptions corresponding to various CUDA driver errors:
class CUDA_ERROR(Exception):
"""CUDA error."""
pass
class CUDA_ERROR_INVALID_VALUE(CUDA_ERROR):
pass
class CUDA_ERROR_OUT_OF_MEMORY(CUDA_ERROR):
pass
class CUDA_ERROR_NOT_INITIALIZED(CUDA_ERROR):
pass
class CUDA_ERROR_DEINITIALIZED(CUDA_ERROR):
pass
class CUDA_ERROR_PROFILER_DISABLED(CUDA_ERROR):
pass
class CUDA_ERROR_PROFILER_NOT_INITIALIZED(CUDA_ERROR):
pass
class CUDA_ERROR_PROFILER_ALREADY_STARTED(CUDA_ERROR):
pass
class CUDA_ERROR_PROFILER_ALREADY_STOPPED(CUDA_ERROR):
pass
class CUDA_ERROR_NO_DEVICE(CUDA_ERROR):
pass
class CUDA_ERROR_INVALID_DEVICE(CUDA_ERROR):
pass
class CUDA_ERROR_INVALID_IMAGE(CUDA_ERROR):
pass
class CUDA_ERROR_INVALID_CONTEXT(CUDA_ERROR):
pass
class CUDA_ERROR_CONTEXT_ALREADY_CURRENT(CUDA_ERROR):
pass
class CUDA_ERROR_MAP_FAILED(CUDA_ERROR):
pass
class CUDA_ERROR_UNMAP_FAILED(CUDA_ERROR):
pass
class CUDA_ERROR_ARRAY_IS_MAPPED(CUDA_ERROR):
pass
class CUDA_ERROR_ALREADY_MAPPED(CUDA_ERROR):
pass
class CUDA_ERROR_NO_BINARY_FOR_GPU(CUDA_ERROR):
pass
class CUDA_ERROR_ALREADY_ACQUIRED(CUDA_ERROR):
pass
class CUDA_ERROR_NOT_MAPPED(CUDA_ERROR):
pass
class CUDA_ERROR_NOT_MAPPED_AS_ARRAY(CUDA_ERROR):
pass
class CUDA_ERROR_NOT_MAPPED_AS_POINTER(CUDA_ERROR):
pass
class CUDA_ERROR_ECC_UNCORRECTABLE(CUDA_ERROR):
pass
class CUDA_ERROR_UNSUPPORTED_LIMIT(CUDA_ERROR):
pass
class CUDA_ERROR_CONTEXT_ALREADY_IN_USE(CUDA_ERROR):
pass
class CUDA_ERROR_PEER_ACCESS_UNSUPPORTED(CUDA_ERROR):
pass
class CUDA_ERROR_INVALID_PTX(CUDA_ERROR):
pass
class CUDA_ERROR_INVALID_GRAPHICS_CONTEXT(CUDA_ERROR):
pass
class CUDA_ERROR_INVALID_SOURCE(CUDA_ERROR):
pass
class CUDA_ERROR_FILE_NOT_FOUND(CUDA_ERROR):
pass
class CUDA_ERROR_SHARED_OBJECT_SYMBOL_NOT_FOUND(CUDA_ERROR):
pass
class CUDA_ERROR_SHARED_OBJECT_INIT_FAILED(CUDA_ERROR):
pass
class CUDA_ERROR_OPERATING_SYSTEM(CUDA_ERROR):
pass
class CUDA_ERROR_INVALID_HANDLE(CUDA_ERROR):
pass
class CUDA_ERROR_NOT_FOUND(CUDA_ERROR):
pass
class CUDA_ERROR_NOT_READY(CUDA_ERROR):
pass
class CUDA_ERROR_ILLEGAL_ADDRESS(CUDA_ERROR):
pass
class CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES(CUDA_ERROR):
pass
class CUDA_ERROR_LAUNCH_TIMEOUT(CUDA_ERROR):
pass
class CUDA_ERROR_LAUNCH_INCOMPATIBLE_TEXTURING(CUDA_ERROR):
pass
class CUDA_ERROR_PEER_ACCESS_ALREADY_ENABLED(CUDA_ERROR):
pass
class CUDA_ERROR_PEER_ACCESS_NOT_ENABLED(CUDA_ERROR):
pass
class CUDA_ERROR_PRIMARY_CONTEXT_ACTIVE(CUDA_ERROR):
pass
class CUDA_ERROR_CONTEXT_IS_DESTROYED(CUDA_ERROR):
pass
class CUDA_ERROR_ASSERT(CUDA_ERROR):
pass
class CUDA_ERROR_TOO_MANY_PEERS(CUDA_ERROR):
pass
class CUDA_ERROR_HOST_MEMORY_ALREADY_REGISTERED(CUDA_ERROR):
pass
class CUDA_ERROR_HOST_MEMORY_NOT_REGISTERED(CUDA_ERROR):
pass
class CUDA_ERROR_HARDWARE_STACK_ERROR(CUDA_ERROR):
pass
class CUDA_ERROR_ILLEGAL_INSTRUCTION(CUDA_ERROR):
pass
class CUDA_ERROR_MISALIGNED_ADDRESS(CUDA_ERROR):
pass
class CUDA_ERROR_INVALID_ADDRESS_SPACE(CUDA_ERROR):
pass
class CUDA_ERROR_INVALID_PC(CUDA_ERROR):
pass
class CUDA_ERROR_LAUNCH_FAILED(CUDA_ERROR):
pass
class CUDA_ERROR_NOT_PERMITTED(CUDA_ERROR):
pass
class CUDA_ERROR_NOT_SUPPORTED(CUDA_ERROR):
pass
class CUDA_ERROR_UNKNOWN(CUDA_ERROR):
pass
CUDA_EXCEPTIONS = {
1: CUDA_ERROR_INVALID_VALUE,
2: CUDA_ERROR_OUT_OF_MEMORY,
3: CUDA_ERROR_NOT_INITIALIZED,
4: CUDA_ERROR_DEINITIALIZED,
5: CUDA_ERROR_PROFILER_DISABLED,
6: CUDA_ERROR_PROFILER_NOT_INITIALIZED,
7: CUDA_ERROR_PROFILER_ALREADY_STARTED,
8: CUDA_ERROR_PROFILER_ALREADY_STOPPED,
100: CUDA_ERROR_NO_DEVICE,
101: CUDA_ERROR_INVALID_DEVICE,
200: CUDA_ERROR_INVALID_IMAGE,
201: CUDA_ERROR_INVALID_CONTEXT,
202: CUDA_ERROR_CONTEXT_ALREADY_CURRENT,
205: CUDA_ERROR_MAP_FAILED,
206: CUDA_ERROR_UNMAP_FAILED,
207: CUDA_ERROR_ARRAY_IS_MAPPED,
208: CUDA_ERROR_ALREADY_MAPPED,
209: CUDA_ERROR_NO_BINARY_FOR_GPU,
210: CUDA_ERROR_ALREADY_ACQUIRED,
211: CUDA_ERROR_NOT_MAPPED,
212: CUDA_ERROR_NOT_MAPPED_AS_ARRAY,
213: CUDA_ERROR_NOT_MAPPED_AS_POINTER,
214: CUDA_ERROR_ECC_UNCORRECTABLE,
215: CUDA_ERROR_UNSUPPORTED_LIMIT,
216: CUDA_ERROR_CONTEXT_ALREADY_IN_USE,
217: CUDA_ERROR_PEER_ACCESS_UNSUPPORTED,
218: CUDA_ERROR_INVALID_PTX,
219: CUDA_ERROR_INVALID_GRAPHICS_CONTEXT,
300: CUDA_ERROR_INVALID_SOURCE,
301: CUDA_ERROR_FILE_NOT_FOUND,
302: CUDA_ERROR_SHARED_OBJECT_SYMBOL_NOT_FOUND,
303: CUDA_ERROR_SHARED_OBJECT_INIT_FAILED,
304: CUDA_ERROR_OPERATING_SYSTEM,
400: CUDA_ERROR_INVALID_HANDLE,
500: CUDA_ERROR_NOT_FOUND,
600: CUDA_ERROR_NOT_READY,
700: CUDA_ERROR_ILLEGAL_ADDRESS,
701: CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES,
702: CUDA_ERROR_LAUNCH_TIMEOUT,
703: CUDA_ERROR_LAUNCH_INCOMPATIBLE_TEXTURING,
704: CUDA_ERROR_PEER_ACCESS_ALREADY_ENABLED,
705: CUDA_ERROR_PEER_ACCESS_NOT_ENABLED,
708: CUDA_ERROR_PRIMARY_CONTEXT_ACTIVE,
709: CUDA_ERROR_CONTEXT_IS_DESTROYED,
710: CUDA_ERROR_ASSERT,
711: CUDA_ERROR_TOO_MANY_PEERS,
712: CUDA_ERROR_HOST_MEMORY_ALREADY_REGISTERED,
713: CUDA_ERROR_HOST_MEMORY_NOT_REGISTERED,
714: CUDA_ERROR_HARDWARE_STACK_ERROR,
715: CUDA_ERROR_ILLEGAL_INSTRUCTION,
716: CUDA_ERROR_MISALIGNED_ADDRESS,
717: CUDA_ERROR_INVALID_ADDRESS_SPACE,
718: CUDA_ERROR_INVALID_PC,
719: CUDA_ERROR_LAUNCH_FAILED,
800: CUDA_ERROR_NOT_PERMITTED,
801: CUDA_ERROR_NOT_SUPPORTED,
999: CUDA_ERROR_UNKNOWN
}
def cuCheckStatus(status):
"""
Raise CUDA exception.
Raise an exception corresponding to the specified CUDA driver
error code.
Parameters
----------
status : int
CUDA driver error code.
See Also
--------
CUDA_EXCEPTIONS
"""
if status != 0:
try:
e = CUDA_EXCEPTIONS[status]
except KeyError:
raise CUDA_ERROR
else:
raise e
CU_POINTER_ATTRIBUTE_CONTEXT = 1
CU_POINTER_ATTRIBUTE_MEMORY_TYPE = 2
CU_POINTER_ATTRIBUTE_DEVICE_POINTER = 3
CU_POINTER_ATTRIBUTE_HOST_POINTER = 4
_libcuda.cuPointerGetAttribute.restype = int
_libcuda.cuPointerGetAttribute.argtypes = [ctypes.c_void_p,
ctypes.c_int,
ctypes.c_uint]
def cuPointerGetAttribute(attribute, ptr):
data = ctypes.c_void_p()
status = _libcuda.cuPointerGetAttribute(data, attribute, ptr)
cuCheckStatus(status)
return data
|
