From c529b182e872b87f264d889b321ea731ee2342fe Mon Sep 17 00:00:00 2001 From: DevKimbob Date: Sun, 30 Mar 2025 19:05:37 +0900 Subject: [PATCH] Fix: Return tuple in get_cuda_version_tuple Changed return type from map object to tuple in get_cuda_version_tuple(). --- bitsandbytes/cuda_specs.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bitsandbytes/cuda_specs.py b/bitsandbytes/cuda_specs.py index ca5463bc4..ff90c50ed 100644 --- a/bitsandbytes/cuda_specs.py +++ b/bitsandbytes/cuda_specs.py @@ -23,9 +23,9 @@ def get_compute_capabilities() -> list[tuple[int, int]]: @lru_cache(None) def get_cuda_version_tuple() -> tuple[int, int]: if torch.version.cuda: - return map(int, torch.version.cuda.split(".")[0:2]) + return tuple(map(int, torch.version.cuda.split(".")[0:2])) elif torch.version.hip: - return map(int, torch.version.hip.split(".")[0:2]) + return tuple(map(int, torch.version.hip.split(".")[0:2])) return None