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