Skip to content

GitLab

Commit bd4a9d7e authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Fix error messages in CUDA script

parent a5f7ec8e
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 7 deletions
scripts/utils.py
View file @ bd4a9d7e
...@@ -81,21 +81,20 @@ def get_nvidia_cc(): ...@@ -81,21 +81,20 @@ def get_nvidia_cc():
result = cuda.cuInit(0) result = cuda.cuInit(0)
if result != CUDA_SUCCESS: if result != CUDA_SUCCESS:
err = cuda.cuGetErrorString(result, ctypes.byref(error_str)) cuda.cuGetErrorString(result, ctypes.byref(error_str))
print(err.value.decode()) return None, error_str.value.decode()
return None, err.value.decode()
result = cuda.cuDeviceGetCount(ctypes.byref(nGpus)) result = cuda.cuDeviceGetCount(ctypes.byref(nGpus))
if result != CUDA_SUCCESS: if result != CUDA_SUCCESS:
err = cuda.cuGetErrorString(result, ctypes.byref(error_str)) cuda.cuGetErrorString(result, ctypes.byref(error_str))
return None, err.value.decode() return None, error_str.value.decode()
if(nGpus.value < 1): if(nGpus.value < 1):
return None, err.value.decode() return None, err.value.decode()
result = cuda.cuDeviceGet(ctypes.byref(device), 0) result = cuda.cuDeviceGet(ctypes.byref(device), 0)
if result != CUDA_SUCCESS: if result != CUDA_SUCCESS:
err = cuda.cuGetErrorString(result, ctypes.byref(error_str)) cuda.cuGetErrorString(result, ctypes.byref(error_str))
return None, err.value.decode() return None, error_str.value.decode()
if cuda.cuDeviceComputeCapability(ctypes.byref(cc_major), ctypes.byref(cc_minor), device) != CUDA_SUCCESS: if cuda.cuDeviceComputeCapability(ctypes.byref(cc_major), ctypes.byref(cc_minor), device) != CUDA_SUCCESS:
return None, "Compute Capability not found" return None, "Compute Capability not found"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment