# -*- python -*- | |
import os | |
cuda_env_vars = [ | |
"CUDA_CACHE_DISABLE", | |
"CUDA_CACHE_MAXSIZE", | |
"CUDA_CACHE_PATH", | |
"CUDA_FORCE_PTX_JIT", | |
"CUDA_VISIBLE_DEVICES", | |
"LD_LIBRARY_PATH", | |
] | |
# We need to propagate cuda environment vars to tests so we can | |
# control how/where the tests get executed. | |
for var in cuda_env_vars: | |
if var in os.environ: | |
config.environment[var] = os.environ[var] | |
config.traditional_output = True |