Evgeny Mankov
277d3b8369
+ Add one matcher (will be more) + Update Maps and Statistics + Add cub_01.cu unit test + Update lit harness to support standalone CUB + Update README.md + Update hipify-perl (only CUB header is supported for now) [IMPORTANT] clang (and hipify-clang) works correctly only with official NVLabs version on GitHub. Compilation of CUB from official CUDA release has conflicts with THRUST. Thus, to compile CUB sources, option "-I" should be specified to the cloned CUB from NVLAB on GitHub.
39 lines
1.6 KiB
Plaintext
39 lines
1.6 KiB
Plaintext
import sys
|
|
import os
|
|
|
|
config.pointer_size = @CMAKE_SIZEOF_VOID_P@
|
|
config.llvm_version = "@LLVM_PACKAGE_VERSION@"
|
|
config.llvm_version_major = int("@LLVM_VERSION_MAJOR@")
|
|
config.llvm_tools_dir = "@LLVM_TOOLS_BINARY_DIR@"
|
|
config.obj_root = "@CMAKE_CURRENT_BINARY_DIR@"
|
|
config.cuda_root = "@CUDA_TOOLKIT_ROOT_DIR@"
|
|
config.cuda_dnn_root = "@CUDA_DNN_ROOT_DIR@"
|
|
config.cuda_cub_root = "@CUDA_CUB_ROOT_DIR@"
|
|
config.cuda_version_major = int("@CUDA_VERSION_MAJOR@")
|
|
config.cuda_version_minor = int("@CUDA_VERSION_MINOR@")
|
|
config.cuda_version = "@CUDA_VERSION@"
|
|
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":
|
|
cuda_version = config.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@"
|
|
if not config.build_type:
|
|
config.build_type = "Debug"
|
|
else:
|
|
config.cuda_sdk_root = config.cuda_root
|
|
|
|
# Support substitution of the tools and libs dirs with user parameters. This is
|
|
# used when we can't determine the tool dir at configuration time.
|
|
try:
|
|
config.llvm_tools_dir = config.llvm_tools_dir % lit_config.params
|
|
config.obj_root = config.obj_root % lit_config.params
|
|
except KeyError:
|
|
e = sys.exc_info()[1]
|
|
key, = e.args
|
|
lit_config.fatal("unable to find %r parameter, use '--param=%s=VALUE'" % (key,key))
|