ecker/bitsandbytes-rocm
1
0
Fork 1
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'main' of github.com:TimDettmers/bitsandbytes into main

Browse Source
This commit is contained in:
Tim Dettmers 2023-07-15 10:22:45 -07:00
commit 37c25c1e0d
2 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
bitsandbytes/__main__.py
View File

@ -98,13 +98,12 @@ generate_bug_report_information()
from . import COMPILED_WITH_CUDA, PACKAGE_GITHUB_URL from . import COMPILED_WITH_CUDA, PACKAGE_GITHUB_URL
from .cuda_setup.env_vars import to_be_ignored 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_header("OTHER")
print(f"COMPILED_WITH_CUDA = {COMPILED_WITH_CUDA}") print(f"COMPILED_WITH_CUDA = {COMPILED_WITH_CUDA}")
cuda = get_cuda_lib_handle() print(f"COMPUTE_CAPABILITIES_PER_GPU = {get_compute_capabilities()}")
print(f"COMPUTE_CAPABILITIES_PER_GPU = {get_compute_capabilities(cuda)}")
print_header("") print_header("")
print_header("DEBUG INFO END") print_header("DEBUG INFO END")
print_header("") print_header("")

2
bitsandbytes/cuda_setup/main.py
View File

@ -87,7 +87,7 @@ class CUDASetup:
make_cmd += '_nomatmul' make_cmd += '_nomatmul'
self.add_log_entry('CUDA SETUP: Something unexpected happened. Please compile from source:') self.add_log_entry('CUDA SETUP: Something unexpected happened. Please compile from source:')
self.add_log_entry('git clone git@github.com:TimDettmers/bitsandbytes.git') self.add_log_entry('git clone https://github.com/TimDettmers/bitsandbytes.git')
self.add_log_entry('cd bitsandbytes') self.add_log_entry('cd bitsandbytes')
self.add_log_entry(make_cmd) self.add_log_entry(make_cmd)
self.add_log_entry('python setup.py install') self.add_log_entry('python setup.py install')