summaryrefslogtreecommitdiff
path: root/bitsandbytes/cuda_setup/main.py
diff options
context:
space:
mode:
authorTim Dettmers <TimDettmers@users.noreply.github.com>2022-09-05 16:29:36 -0700
committerGitHub <noreply@github.com>2022-09-05 16:29:36 -0700
commitf0ae860c86039d1c1e41166aaf2153a5bd9b9a89 (patch)
tree75477acfbce2da2a753ee21d4cf0da64f3f50ea5 /bitsandbytes/cuda_setup/main.py
parenteab4d8232d558f2e6bd7f7cc3d00e2e6e94f4e80 (diff)
parentaca55881b9815a462142f42f3ff0dc917830d85c (diff)
Merge pull request #25 from TimDettmers/remove_unused_code
Remove unused code, switch to warnings
Diffstat (limited to 'bitsandbytes/cuda_setup/main.py')
-rw-r--r--bitsandbytes/cuda_setup/main.py4
1 files changed, 0 insertions, 4 deletions
diff --git a/bitsandbytes/cuda_setup/main.py b/bitsandbytes/cuda_setup/main.py
index d305c64..78a2844 100644
--- a/bitsandbytes/cuda_setup/main.py
+++ b/bitsandbytes/cuda_setup/main.py
@@ -17,10 +17,7 @@ evaluation:
"""
import ctypes
-import torch
-from pathlib import Path
-from ..utils import execute_and_return
from .paths import determine_cuda_runtime_lib_path
@@ -81,7 +78,6 @@ def get_compute_capabilities(cuda):
cc_major = ctypes.c_int()
cc_minor = ctypes.c_int()
- result = ctypes.c_int()
device = ctypes.c_int()
check_cuda_result(cuda, cuda.cuDeviceGetCount(ctypes.byref(nGpus)))