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
ce126d46
Commit
ce126d46
authored
Jul 15, 2023
by
ihsanturk
Browse files
deleted references to get_cuda_lib_handle
parent
2f0f0e5d
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
14 deletions
+3
-14
bitsandbytes/__main__.py
bitsandbytes/__main__.py
+2
-3
bitsandbytes/cuda_setup/main.py
bitsandbytes/cuda_setup/main.py
+1
-11
No files found.
bitsandbytes/__main__.py
View file @
ce126d46
...
...
@@ -97,13 +97,12 @@ generate_bug_report_information()
from
.
import
COMPILED_WITH_CUDA
,
PACKAGE_GITHUB_URL
from
.cuda_setup.env_vars
import
to_be_ignored
from
.cuda_setup.main
import
get_compute_capabilities
,
get_cuda_lib_handle
from
.cuda_setup.main
import
get_compute_capabilities
print_header
(
"OTHER"
)
print
(
f
"COMPILED_WITH_CUDA =
{
COMPILED_WITH_CUDA
}
"
)
cuda
=
get_cuda_lib_handle
()
print
(
f
"COMPUTE_CAPABILITIES_PER_GPU =
{
get_compute_capabilities
(
cuda
)
}
"
)
print
(
f
"COMPUTE_CAPABILITIES_PER_GPU =
{
get_compute_capabilities
()
}
"
)
print_header
(
""
)
print_header
(
"DEBUG INFO END"
)
print_header
(
""
)
...
...
bitsandbytes/cuda_setup/main.py
View file @
ce126d46
...
...
@@ -359,14 +359,4 @@ def evaluate_cuda_setup():
"if not has_cublaslt (CC < 7.5), then we have to choose _nocublaslt.so"
binary_name
=
f
"libbitsandbytes_cuda
{
cuda_version_string
}
_nocublaslt.so"
return
binary_name
,
cudart_path
,
cc
,
cuda_version_string
def
get_cuda_lib_handle
():
# 1. find libcuda.so library (GPU driver) (/usr/lib)
try
:
cuda
=
ct
.
CDLL
(
"libcuda.so"
)
except
OSError
:
CUDASetup
.
get_instance
().
add_log_entry
(
'CUDA SETUP: WARNING! libcuda.so not found! Do you have a CUDA driver installed? If you are on a cluster, make sure you are on a CUDA machine!'
)
return
None
return
cuda
\ No newline at end of file
return
binary_name
,
cudart_path
,
cc
,
cuda_version_string
\ No newline at end of file
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