Merge pull request #315 from emankov/win_testing

[HIPIFY][tests][win] CUDA samples root env. var is changed
This commit is contained in:
Evgeny Mankov
2018-01-14 19:18:07 +03:00
gecommit door GitHub
bovenliggende 0953fa4a6b 39a0372077
commit a42671d133
@@ -7,8 +7,10 @@ config.cuda_root = "@CUDA_TOOLKIT_ROOT_DIR@"
if sys.platform in ['win32']:
config.cuda_sdk_root = "@CUDA_SDK_ROOT_DIR@"
if not config.cuda_sdk_root or config.cuda_sdk_root == "CUDA_SDK_ROOT_DIR-NOTFOUND":
config.cuda_samples_root = os.environ.get('NVCUDASAMPLES_ROOT')
if not config.cuda_samples_root or config.cuda_samples_root == "NVCUDASAMPLES_ROOT-NOTFOUND":
cuda_version = "@CUDA_VERSION@"
cuda_version = cuda_version.replace('.','_')
config.cuda_samples_root = os.environ.get('NVCUDASAMPLES' + cuda_version + '_ROOT')
if not config.cuda_samples_root:
lit_config.fatal('No CUDA Samples dir set! Please set CUDA_SDK_ROOT_DIR.')
config.cuda_sdk_root = config.cuda_samples_root
config.build_type = "@CMAKE_BUILD_TYPE@"