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