1@LIT_SITE_CFG_IN_HEADER@ 2 3import sys 4 5config.target_triple = "@LLVM_TARGET_TRIPLE@" 6config.llvm_src_root = "@LLVM_SOURCE_DIR@" 7config.llvm_tools_dir = lit_config.substitute("@LLVM_TOOLS_DIR@") 8config.llvm_shlib_ext = "@SHLIBEXT@" 9config.llvm_shlib_dir = lit_config.substitute(path(r"@SHLIBDIR@")) 10config.python_executable = "@Python3_EXECUTABLE@" 11config.enable_assertions = @ENABLE_ASSERTIONS@ 12config.native_target = "@LLVM_NATIVE_ARCH@" 13config.host_os = "@HOST_OS@" 14config.host_cc = "@HOST_CC@" 15config.host_cxx = "@HOST_CXX@" 16config.enable_libcxx = "@LLVM_ENABLE_LIBCXX@" 17config.host_cmake = "@CMAKE_COMMAND@" 18config.host_cmake_generator = "@CMAKE_GENERATOR@" 19config.llvm_use_linker = "@LLVM_USE_LINKER@" 20config.llvm_use_sanitizer = "@LLVM_USE_SANITIZER@" 21config.host_arch = "@HOST_ARCH@" 22config.mlir_src_root = "@MLIR_SOURCE_DIR@" 23config.mlir_obj_root = "@MLIR_BINARY_DIR@" 24config.mlir_tools_dir = "@MLIR_TOOLS_DIR@" 25config.mlir_cmake_dir = "@MLIR_CMAKE_DIR@" 26config.mlir_lib_dir = "@MLIR_LIB_DIR@" 27 28config.build_examples = @LLVM_BUILD_EXAMPLES@ 29config.run_nvptx_tests = @LLVM_HAS_NVPTX_TARGET@ 30config.enable_cuda_runner = @MLIR_ENABLE_CUDA_RUNNER@ 31config.run_rocm_tests = @MLIR_ENABLE_ROCM_CONVERSIONS@ 32config.enable_rocm_runner = @MLIR_ENABLE_ROCM_RUNNER@ 33config.gpu_compilation_format = "@MLIR_GPU_COMPILATION_TEST_FORMAT@" 34config.rocm_test_chipset = "@ROCM_TEST_CHIPSET@" 35config.enable_sycl_runner = @MLIR_ENABLE_SYCL_RUNNER@ 36config.enable_spirv_cpu_runner = @MLIR_ENABLE_SPIRV_CPU_RUNNER@ 37config.enable_vulkan_runner = @MLIR_ENABLE_VULKAN_RUNNER@ 38config.enable_bindings_python = @MLIR_ENABLE_BINDINGS_PYTHON@ 39config.intel_sde_executable = "@INTEL_SDE_EXECUTABLE@" 40config.mlir_run_amx_tests = @MLIR_RUN_AMX_TESTS@ 41config.mlir_run_arm_sve_tests = @MLIR_RUN_ARM_SVE_TESTS@ 42# This is a workaround for the fact that LIT's: 43# %if <cond> 44# requires <cond> to be in the set of available features. 45# TODO: Update LIT's TestRunner so that this is not required. 46if config.mlir_run_arm_sve_tests: 47 config.available_features.add("mlir_arm_sve_tests") 48config.mlir_run_arm_sme_tests = @MLIR_RUN_ARM_SME_TESTS@ 49config.mlir_run_x86vector_tests = @MLIR_RUN_X86VECTOR_TESTS@ 50config.mlir_run_riscv_vector_tests = "@MLIR_RUN_RISCV_VECTOR_TESTS@" 51config.mlir_run_cuda_tensor_core_tests = @MLIR_RUN_CUDA_TENSOR_CORE_TESTS@ 52config.mlir_run_cuda_sm80_tests = @MLIR_RUN_CUDA_SM80_TESTS@ 53config.mlir_run_cuda_sm80_lt_tests = @MLIR_RUN_CUDA_SM80_LT_TESTS@ 54config.mlir_run_cuda_sm90_tests = @MLIR_RUN_CUDA_SM90_TESTS@ 55config.mlir_include_integration_tests = @MLIR_INCLUDE_INTEGRATION_TESTS@ 56config.arm_emulator_executable = "@ARM_EMULATOR_EXECUTABLE@" 57# Some tests marked with 'UNSUPPORTED: target=aarch64{{.*}}' are still run when 58# configured with ARM_EMULATOR_EXECUTABLE and the default target is not aarch64. 59# Add feature 'mlir_arm_emulator' here and to UNSUPPORTED list of such tests. 60if config.arm_emulator_executable: 61 config.available_features.add("mlir_arm_emulator") 62config.arm_emulator_options = "@ARM_EMULATOR_OPTIONS@" 63config.arm_emulator_mlir_runner_executable = "@ARM_EMULATOR_MLIR_RUNNER_EXECUTABLE@" 64config.arm_emulator_lli_executable = "@ARM_EMULATOR_LLI_EXECUTABLE@" 65config.arm_emulator_utils_lib_dir = "@ARM_EMULATOR_UTILS_LIB_DIR@" 66config.arm_sme_abi_routines_shlib = "@ARM_SME_ABI_ROUTINES_SHLIB@" 67config.riscv_vector_emulator_executable = "@RISCV_VECTOR_EMULATOR_EXECUTABLE@" 68config.riscv_vector_emulator_options = "@RISCV_VECTOR_EMULATOR_OPTIONS@" 69config.riscv_emulator_lli_executable = "@RISCV_EMULATOR_LLI_EXECUTABLE@" 70config.riscv_emulator_utils_lib_dir = "@RISCV_EMULATOR_UTILS_LIB_DIR@" 71 72import lit.llvm 73lit.llvm.initialize(lit_config, config) 74 75# Let the main config do the real work. 76lit_config.load_config(config, "@MLIR_SOURCE_DIR@/test/lit.cfg.py") 77