Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
bitsandbytes
Commits
b3de1921
"tests/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "e4e0132972a09df8f73387c4ea48d2cd19178d2a"
Commit
b3de1921
authored
Jan 03, 2023
by
Tim Dettmers
Browse files
Added error message for unexpected CUDA exception.
parent
81990491
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
1 deletion
+4
-1
bitsandbytes/cuda_setup/main.py
bitsandbytes/cuda_setup/main.py
+4
-1
No files found.
bitsandbytes/cuda_setup/main.py
View file @
b3de1921
...
@@ -268,7 +268,10 @@ def check_cuda_result(cuda, result_val):
...
@@ -268,7 +268,10 @@ def check_cuda_result(cuda, result_val):
if
result_val
!=
0
:
if
result_val
!=
0
:
error_str
=
ct
.
c_char_p
()
error_str
=
ct
.
c_char_p
()
cuda
.
cuGetErrorString
(
result_val
,
ct
.
byref
(
error_str
))
cuda
.
cuGetErrorString
(
result_val
,
ct
.
byref
(
error_str
))
CUDASetup
.
get_instance
().
add_log_entry
(
f
"CUDA exception! Error code:
{
error_str
.
value
.
decode
()
}
"
)
if
error_str
.
value
is
not
None
:
CUDASetup
.
get_instance
().
add_log_entry
(
f
"CUDA exception! Error code:
{
error_str
.
value
.
decode
()
}
"
)
else
:
CUDASetup
.
get_instance
().
add_log_entry
(
f
"Unknown CUDA exception! Please check your CUDA install. It might also be that your GPU is too old."
)
# https://docs.nvidia.com/cuda/cuda-runtime-api/group__CUDART____VERSION.html#group__CUDART____VERSION
# https://docs.nvidia.com/cuda/cuda-runtime-api/group__CUDART____VERSION.html#group__CUDART____VERSION
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment