diff --git a/sycl/test-e2e/lit.cfg.py b/sycl/test-e2e/lit.cfg.py index c0e9f9c926c20..1ba4d04acb853 100644 --- a/sycl/test-e2e/lit.cfg.py +++ b/sycl/test-e2e/lit.cfg.py @@ -6,6 +6,7 @@ import re import subprocess import textwrap +import shlex import shutil import lit.formats @@ -304,12 +305,18 @@ def open_check_file(file_name): file=fp, ) -config.level_zero_libs_dir = lit_config.params.get( - "level_zero_libs_dir", config.level_zero_libs_dir +config.level_zero_libs_dir = shlex.quote( + lit_config.params.get("level_zero_libs_dir", config.level_zero_libs_dir) ) -config.level_zero_include = lit_config.params.get( - "level_zero_include", - (config.level_zero_include if config.level_zero_include else config.sycl_include), +config.level_zero_include = shlex.quote( + lit_config.params.get( + "level_zero_include", + ( + config.level_zero_include + if config.level_zero_include + else config.sycl_include + ), + ) ) level_zero_options = level_zero_options = ( @@ -409,10 +416,14 @@ def open_check_file(file_name): file=fp, ) -config.cuda_libs_dir = lit_config.params.get("cuda_libs_dir", config.cuda_libs_dir) -config.cuda_include = lit_config.params.get( - "cuda_include", - (config.cuda_include if config.cuda_include else config.sycl_include), +config.cuda_libs_dir = shlex.quote( + lit_config.params.get("cuda_libs_dir", config.cuda_libs_dir) +) +config.cuda_include = shlex.quote( + lit_config.params.get( + "cuda_include", + (config.cuda_include if config.cuda_include else config.sycl_include), + ) ) cuda_options = cuda_options = ( @@ -454,10 +465,14 @@ def open_check_file(file_name): ), file=fp, ) -config.hip_libs_dir = lit_config.params.get("hip_libs_dir", config.hip_libs_dir) -config.hip_include = lit_config.params.get( - "hip_include", - (config.hip_include if config.hip_include else config.sycl_include), +config.hip_libs_dir = shlex.quote( + lit_config.params.get("hip_libs_dir", config.hip_libs_dir) +) +config.hip_include = shlex.quote( + lit_config.params.get( + "hip_include", + (config.hip_include if config.hip_include else config.sycl_include), + ) ) hip_options = hip_options = (