-
Notifications
You must be signed in to change notification settings - Fork 181
/
lit.site.cfg.py.in
51 lines (42 loc) · 2.13 KB
/
lit.site.cfg.py.in
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
@LIT_SITE_CFG_IN_HEADER@
# ============================================================================ #
# Copyright (c) 2022 - 2024 NVIDIA Corporation & Affiliates. #
# All rights reserved. #
# #
# This source code and the accompanying materials are made available under #
# the terms of the Apache License 2.0 which accompanies this distribution. #
# ============================================================================ #
import sys
# Helper function to convert cmake variable boolean definitions to True/False.
def cmake_boolvar_to_bool(s):
return s == "1"
config.llvm_tools_dir = lit_config.substitute("@LLVM_TOOLS_BINARY_DIR@")
config.llvm_shlib_dir = lit_config.substitute(path(r"@SHLIBDIR@"))
config.llvm_plugin_ext = "@LLVM_PLUGIN_EXT@"
config.llvm_install = config.llvm_tools_dir + "/.."
config.lit_tools_dir = "@LLVM_LIT_TOOLS_DIR@"
config.errc_messages = "@LLVM_LIT_ERRC_MESSAGES@"
config.cudaq_obj_root = "@CUDAQ_BINARY_DIR@"
config.cudaq_src_dir = "@CUDAQ_SOURCE_DIR@"
config.cudaq_target_dir = "@CMAKE_BINARY_DIR@/targets"
config.cudaq_tools_dir = lit_config.substitute("@CUDAQ_TOOLS_DIR@")
config.cudaq_intrinsic_modules_dir = "@CUDAQ_INTRINSIC_MODULES_DIR@"
config.cudaq_llvm_tools_dir = "@CMAKE_BINARY_DIR@/bin"
config.cudaq_lib_dir = "@CMAKE_BINARY_DIR@/lib"
config.cudaq_plugin_ext = "@LLVM_PLUGIN_EXT@"
config.python_executable = "@PYTHON_EXECUTABLE@"
config.cc = "@CMAKE_C_COMPILER@"
config.targets_to_build = "@TARGETS_TO_BUILD@"
config.excludes = ["@CUDAQ_EXTRA_CONFIG_EXCLUDES@"]
config.pipefail = False
# Is the nvcc compiler available?
config.cuda_found = "@CUDA_FOUND@"
if cmake_boolvar_to_bool(config.cuda_found):
config.available_features.add('nvcc')
config.test_remote_sim = "@CUDAQ_TEST_REMOTE_SIM@"
if cmake_boolvar_to_bool(config.test_remote_sim):
config.available_features.add('remote-sim')
import lit.llvm
lit.llvm.initialize(lit_config, config)
# Let the main config do the real work.
lit_config.load_config(config, "@CUDAQ_SOURCE_DIR@/targettests/lit.cfg.py")