# -*- python -*-

import os

cuda_env_vars = [
    'CUDA_CACHE_DISABLE',
    'CUDA_CACHE_MAXSIZE',
    'CUDA_CACHE_PATH',
    'CUDA_FORCE_PTX_JIT',
    'CUDA_VISIBLE_DEVICES',
]

# 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

# "compiletime" wants to traverse whole build directory after each test which
# takes a lot of time if we have config with split thrust tests. Disable it
# until it has smaller overhead.
config.test_modules.remove("compiletime")