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