|
6 | 6 | import re |
7 | 7 | import subprocess |
8 | 8 | import textwrap |
| 9 | +import shlex |
9 | 10 | import shutil |
10 | 11 |
|
11 | 12 | import lit.formats |
@@ -304,13 +305,13 @@ def open_check_file(file_name): |
304 | 305 | file=fp, |
305 | 306 | ) |
306 | 307 |
|
307 | | -config.level_zero_libs_dir = lit_config.params.get( |
| 308 | +config.level_zero_libs_dir = shlex.quote(lit_config.params.get( |
308 | 309 | "level_zero_libs_dir", config.level_zero_libs_dir |
309 | | -) |
310 | | -config.level_zero_include = lit_config.params.get( |
| 310 | +)) |
| 311 | +config.level_zero_include = shlex.quote(lit_config.params.get( |
311 | 312 | "level_zero_include", |
312 | 313 | (config.level_zero_include if config.level_zero_include else config.sycl_include), |
313 | | -) |
| 314 | +)) |
314 | 315 |
|
315 | 316 | level_zero_options = level_zero_options = ( |
316 | 317 | (" -L" + config.level_zero_libs_dir if config.level_zero_libs_dir else "") |
@@ -409,11 +410,11 @@ def open_check_file(file_name): |
409 | 410 | file=fp, |
410 | 411 | ) |
411 | 412 |
|
412 | | -config.cuda_libs_dir = lit_config.params.get("cuda_libs_dir", config.cuda_libs_dir) |
413 | | -config.cuda_include = lit_config.params.get( |
| 413 | +config.cuda_libs_dir = shlex.quote(lit_config.params.get("cuda_libs_dir", config.cuda_libs_dir)) |
| 414 | +config.cuda_include = shlex.quote(lit_config.params.get( |
414 | 415 | "cuda_include", |
415 | 416 | (config.cuda_include if config.cuda_include else config.sycl_include), |
416 | | -) |
| 417 | +)) |
417 | 418 |
|
418 | 419 | cuda_options = cuda_options = ( |
419 | 420 | (" -L" + config.cuda_libs_dir if config.cuda_libs_dir else "") |
@@ -454,11 +455,11 @@ def open_check_file(file_name): |
454 | 455 | ), |
455 | 456 | file=fp, |
456 | 457 | ) |
457 | | -config.hip_libs_dir = lit_config.params.get("hip_libs_dir", config.hip_libs_dir) |
458 | | -config.hip_include = lit_config.params.get( |
| 458 | +config.hip_libs_dir = shlex.quote(lit_config.params.get("hip_libs_dir", config.hip_libs_dir)) |
| 459 | +config.hip_include = shlex.quote(lit_config.params.get( |
459 | 460 | "hip_include", |
460 | 461 | (config.hip_include if config.hip_include else config.sycl_include), |
461 | | -) |
| 462 | +)) |
462 | 463 |
|
463 | 464 | hip_options = hip_options = ( |
464 | 465 | (" -L" + config.hip_libs_dir if config.hip_libs_dir else "") |
|
0 commit comments