"tools/cfgs/git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "83954d039e48aedc95b09dac2075eb38f571d326"
Commit 5820c750 authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

Fix get_nvidia_cc to handle cases where cuda.cuGetErrorString gives a None result

parent baf15f8f
...@@ -82,13 +82,16 @@ def get_nvidia_cc(): ...@@ -82,13 +82,16 @@ def get_nvidia_cc():
result = cuda.cuInit(0) result = cuda.cuInit(0)
if result != CUDA_SUCCESS: if result != CUDA_SUCCESS:
cuda.cuGetErrorString(result, ctypes.byref(error_str)) cuda.cuGetErrorString(result, ctypes.byref(error_str))
return None, error_str.value.decode() if error_str.value:
return None, error_str.value.decode()
else:
return None, "Unknown error: cuInit returned %d" % result
result = cuda.cuDeviceGetCount(ctypes.byref(nGpus)) result = cuda.cuDeviceGetCount(ctypes.byref(nGpus))
if result != CUDA_SUCCESS: if result != CUDA_SUCCESS:
cuda.cuGetErrorString(result, ctypes.byref(error_str)) cuda.cuGetErrorString(result, ctypes.byref(error_str))
return None, error_str.value.decode() return None, error_str.value.decode()
if(nGpus.value < 1): if nGpus.value < 1:
return None, "No GPUs detected" return None, "No GPUs detected"
result = cuda.cuDeviceGet(ctypes.byref(device), 0) result = cuda.cuDeviceGet(ctypes.byref(device), 0)
......
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