blob: 4a605063347a2595549ebe5b4f2a2f314e2d6758 [file] [log] [blame]
# -*- 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