From e38b9e91b718e8b84f4678c423f72dd4decce4e5 Mon Sep 17 00:00:00 2001 From: arlo-phoenix Date: Tue, 8 Aug 2023 21:31:20 +0200 Subject: [PATCH] Revert get_cuda_version ROCM version change not called anymore --- bitsandbytes/cuda_setup/main.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/bitsandbytes/cuda_setup/main.py b/bitsandbytes/cuda_setup/main.py index ec99fb7..c81e960 100644 --- a/bitsandbytes/cuda_setup/main.py +++ b/bitsandbytes/cuda_setup/main.py @@ -308,14 +308,10 @@ def determine_cuda_runtime_lib_path() -> Union[Path, None]: # https://docs.nvidia.com/cuda/cuda-runtime-api/group__CUDART____VERSION.html#group__CUDART____VERSION def get_cuda_version(): - if torch.version.cuda: - major, minor = map(int, torch.version.cuda.split(".")) + major, minor = map(int, torch.version.cuda.split(".")) if major < 11: CUDASetup.get_instance().add_log_entry('CUDA SETUP: CUDA version lower than 11 are currently not supported for LLM.int8(). You will be only to use 8-bit optimizers and quantization routines!!') - elif torch.version.hip: - major, minor = map(int, torch.version.hip.split(".")) - return f'{major}{minor}'