From dd50382b3237897d336f2c8f7b7f555b3c851abb Mon Sep 17 00:00:00 2001 From: Tim Dettmers Date: Sun, 31 Jul 2022 17:47:44 -0700 Subject: Full evaluate_cuda setup with integration test. --- tests/test_cuda_setup_evaluator.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) (limited to 'tests/test_cuda_setup_evaluator.py') diff --git a/tests/test_cuda_setup_evaluator.py b/tests/test_cuda_setup_evaluator.py index 96ee6c5..72aa3c7 100644 --- a/tests/test_cuda_setup_evaluator.py +++ b/tests/test_cuda_setup_evaluator.py @@ -1,4 +1,5 @@ import pytest +import os from typing import List @@ -16,6 +17,7 @@ HAPPY_PATH__LD_LIB_TEST_PATHS: List[tuple[str,str]] = [ (f"some/other/dir:dir/with/{CUDA_RUNTIME_LIB}:", f"dir/with/{CUDA_RUNTIME_LIB}"), (f"some/other/dir::dir/with/{CUDA_RUNTIME_LIB}", f"dir/with/{CUDA_RUNTIME_LIB}"), (f"dir/with/{CUDA_RUNTIME_LIB}:some/other/dir", f"dir/with/{CUDA_RUNTIME_LIB}"), + (f"dir/with/{CUDA_RUNTIME_LIB}:other/dir/libcuda.so", f"dir/with/{CUDA_RUNTIME_LIB}"), ] @@ -64,3 +66,21 @@ def test_get_cuda_runtime_lib_path__non_existent_dir(capsys, tmp_path): match in std_err for match in {"WARNING", "non-existent"} ) + +def test_full_system(): + ## this only tests the cuda version and not compute capability + ld_path = os.environ['LD_LIBRARY_PATH'] + paths = ld_path.split(':') + version = '' + for p in paths: + if 'cuda' in p: + idx = p.rfind('cuda-') + version = p[idx+5:idx+5+4].replace('/', '') + version = float(version) + break + + binary_name = evaluate_cuda_setup() + binary_name = binary_name.replace('libbitsandbytes_cuda', '') + assert binary_name.startswith(str(version).replace('.', '')) + + -- cgit v1.2.3