We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1da7332 commit c3cfcbcCopy full SHA for c3cfcbc
onnx_tensorrt/backend.py
@@ -18,6 +18,8 @@ def cudaSetDevice(device_idx):
18
ret = libcudart.cudaSetDevice(device_idx)
19
if ret != 0:
20
error_string = libcudart.cudaGetErrorString(ret)
21
+ if isinstance(error_string, bytes):
22
+ error_string = error_string.decode("utf-8")
23
raise RuntimeError("cudaSetDevice: " + error_string)
24
25
def count_trailing_ones(vals):
0 commit comments