# -*- 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 |