From 663a9d2ff6300d990aabb96e00d8b24505ed3eec Mon Sep 17 00:00:00 2001 From: m moadeli Date: Mon, 5 Feb 2024 15:20:11 +0000 Subject: [PATCH] Include the necessary environment paths during the test-e2e build process for CUDA or HIP backends. The absence of the added path leads to the inability to locate libdevice for specific architectures, resulting in a failure. --- sycl/test-e2e/lit.cfg.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/sycl/test-e2e/lit.cfg.py b/sycl/test-e2e/lit.cfg.py index e1b2b4a1ef28d..2729eb204ae2c 100644 --- a/sycl/test-e2e/lit.cfg.py +++ b/sycl/test-e2e/lit.cfg.py @@ -80,6 +80,12 @@ llvm_config.with_environment("PATH", config.lit_tools_dir, append_path=True) +if "cuda:gpu" in config.sycl_devices: + llvm_config.with_system_environment("CUDA_PATH") + +if "hip:gpu" in config.sycl_devices: + llvm_config.with_system_environment("ROCM_PATH") + # Configure LD_LIBRARY_PATH or corresponding os-specific alternatives if platform.system() == "Linux": config.available_features.add("linux")