summaryrefslogtreecommitdiff
path: root/bitsandbytes/cuda_setup
diff options
context:
space:
mode:
authorTim Dettmers <tim.dettmers@gmail.com>2022-10-25 08:01:19 -0700
committerTim Dettmers <tim.dettmers@gmail.com>2022-10-25 08:01:19 -0700
commita371be302ddbdf3f36acef1a6fe365672099c9d9 (patch)
treee86da0916db9bf6184ce4cc06ebd9ebfeb2762a2 /bitsandbytes/cuda_setup
parent62e1649357095473fe6055c6f6b03743d1753e11 (diff)
Added CUDA SETUP instruction generator.
Diffstat (limited to 'bitsandbytes/cuda_setup')
-rw-r--r--bitsandbytes/cuda_setup/main.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/bitsandbytes/cuda_setup/main.py b/bitsandbytes/cuda_setup/main.py
index f8f35f0..f5abda2 100644
--- a/bitsandbytes/cuda_setup/main.py
+++ b/bitsandbytes/cuda_setup/main.py
@@ -159,4 +159,4 @@ def evaluate_cuda_setup():
binary_name = get_binary_name()
- return binary_name
+ return binary_name, cudart_path, cuda, cc, cuda_version_string