Follow the same pattern as all the other test suites where the PATHS option to configure_lit_site_cfg and the path() function in the lit site config are used to make site configs relocatable.
87 lines
4.3 KiB
Python
87 lines
4.3 KiB
Python
@LIT_SITE_CFG_IN_HEADER@
|
|
|
|
import sys
|
|
|
|
config.target_triple = "@LLVM_TARGET_TRIPLE@"
|
|
config.llvm_src_root = path(r"@LLVM_SOURCE_DIR@")
|
|
config.llvm_tools_dir = lit_config.substitute(path(r"@LLVM_TOOLS_DIR@"))
|
|
config.lit_tools_dir = path(r"@LLVM_LIT_TOOLS_DIR@")
|
|
config.spirv_tools_tests = @LLVM_INCLUDE_SPIRV_TOOLS_TESTS@
|
|
config.llvm_shlib_ext = "@SHLIBEXT@"
|
|
config.llvm_shlib_dir = lit_config.substitute(path(r"@SHLIBDIR@"))
|
|
config.python_executable = "@Python3_EXECUTABLE@"
|
|
config.enable_assertions = @ENABLE_ASSERTIONS@
|
|
config.expensive_checks = "@EXPENSIVE_CHECKS@"
|
|
config.native_target = "@LLVM_NATIVE_ARCH@"
|
|
config.host_os = "@HOST_OS@"
|
|
config.host_cc = "@HOST_CC@"
|
|
config.host_cxx = "@HOST_CXX@"
|
|
config.host_c_compiler_launcher = "@CMAKE_C_COMPILER_LAUNCHER@"
|
|
config.host_cxx_compiler_launcher = "@CMAKE_CXX_COMPILER_LAUNCHER@"
|
|
config.enable_libcxx = "@LLVM_ENABLE_LIBCXX@"
|
|
config.host_cmake = "@CMAKE_COMMAND@"
|
|
config.host_cmake_generator = "@CMAKE_GENERATOR@"
|
|
config.cmake_build_type = "@CMAKE_BUILD_TYPE@"
|
|
config.llvm_use_linker = "@LLVM_USE_LINKER@"
|
|
config.llvm_use_sanitizer = "@LLVM_USE_SANITIZER@"
|
|
config.host_arch = "@HOST_ARCH@"
|
|
config.mlir_src_root = path(r"@MLIR_SOURCE_DIR@")
|
|
config.mlir_obj_root = path(r"@MLIR_BINARY_DIR@")
|
|
config.mlir_tools_dir = path(r"@MLIR_TOOLS_DIR@")
|
|
config.mlir_cmake_dir = path(r"@MLIR_CMAKE_DIR@")
|
|
config.mlir_lib_dir = path(r"@MLIR_LIB_DIR@")
|
|
|
|
config.build_examples = @LLVM_BUILD_EXAMPLES@
|
|
config.run_nvptx_tests = @LLVM_HAS_NVPTX_TARGET@
|
|
config.enable_cuda_runner = @MLIR_ENABLE_CUDA_RUNNER@
|
|
config.run_rocm_tests = @LLVM_HAS_AMDGPU_TARGET@
|
|
config.enable_rocm_runner = @MLIR_ENABLE_ROCM_RUNNER@
|
|
config.gpu_compilation_format = "@MLIR_GPU_COMPILATION_TEST_FORMAT@"
|
|
config.rocm_test_chipset = "@ROCM_TEST_CHIPSET@"
|
|
config.run_xevm_tests = @MLIR_ENABLE_XEVM_CONVERSIONS@
|
|
config.enable_sycl_runner = @MLIR_ENABLE_SYCL_RUNNER@
|
|
config.enable_levelzero_runner = @MLIR_ENABLE_LEVELZERO_RUNNER@
|
|
config.enable_spirv_cpu_runner = @MLIR_ENABLE_SPIRV_CPU_RUNNER@
|
|
config.enable_vulkan_runner = @MLIR_ENABLE_VULKAN_RUNNER@
|
|
config.enable_bindings_python = @MLIR_ENABLE_BINDINGS_PYTHON@
|
|
config.enable_python_stable_abi = @MLIR_ENABLE_PYTHON_STABLE_ABI@
|
|
config.intel_sde_executable = "@INTEL_SDE_EXECUTABLE@"
|
|
config.mlir_run_x86_amx_tests = @MLIR_RUN_X86_AMX_TESTS@
|
|
config.mlir_run_arm_sve_tests = @MLIR_RUN_ARM_SVE_TESTS@
|
|
# This is a workaround for the fact that LIT's:
|
|
# %if <cond>
|
|
# requires <cond> to be in the set of available features.
|
|
# TODO: Update LIT's TestRunner so that this is not required.
|
|
if config.mlir_run_arm_sve_tests:
|
|
config.available_features.add("mlir_arm_sve_tests")
|
|
config.mlir_run_arm_sme_tests = @MLIR_RUN_ARM_SME_TESTS@
|
|
config.mlir_run_x86_tests = @MLIR_RUN_X86_TESTS@
|
|
config.mlir_run_riscv_vector_tests = "@MLIR_RUN_RISCV_VECTOR_TESTS@"
|
|
config.mlir_run_cuda_tensor_core_tests = @MLIR_RUN_CUDA_TENSOR_CORE_TESTS@
|
|
config.mlir_run_cuda_sm80_tests = @MLIR_RUN_CUDA_SM80_TESTS@
|
|
config.mlir_run_cuda_sm80_lt_tests = @MLIR_RUN_CUDA_SM80_LT_TESTS@
|
|
config.mlir_run_cuda_sm90_tests = @MLIR_RUN_CUDA_SM90_TESTS@
|
|
config.mlir_include_integration_tests = @MLIR_INCLUDE_INTEGRATION_TESTS@
|
|
config.llvm_shared_libs_build = @BUILD_SHARED_LIBS@
|
|
config.arm_emulator_executable = "@ARM_EMULATOR_EXECUTABLE@"
|
|
# Some tests marked with 'UNSUPPORTED: target=aarch64{{.*}}' are still run when
|
|
# configured with ARM_EMULATOR_EXECUTABLE and the default target is not aarch64.
|
|
# Add feature 'mlir_arm_emulator' here and to UNSUPPORTED list of such tests.
|
|
if config.arm_emulator_executable:
|
|
config.available_features.add("mlir_arm_emulator")
|
|
config.arm_emulator_options = "@ARM_EMULATOR_OPTIONS@"
|
|
config.arm_emulator_mlir_runner_executable = "@ARM_EMULATOR_MLIR_RUNNER_EXECUTABLE@"
|
|
config.arm_emulator_lli_executable = "@ARM_EMULATOR_LLI_EXECUTABLE@"
|
|
config.arm_emulator_utils_lib_dir = "@ARM_EMULATOR_UTILS_LIB_DIR@"
|
|
config.arm_sme_abi_routines_shlib = "@ARM_SME_ABI_ROUTINES_SHLIB@"
|
|
config.riscv_vector_emulator_executable = "@RISCV_VECTOR_EMULATOR_EXECUTABLE@"
|
|
config.riscv_vector_emulator_options = "@RISCV_VECTOR_EMULATOR_OPTIONS@"
|
|
config.riscv_emulator_lli_executable = "@RISCV_EMULATOR_LLI_EXECUTABLE@"
|
|
config.riscv_emulator_utils_lib_dir = "@RISCV_EMULATOR_UTILS_LIB_DIR@"
|
|
|
|
import lit.llvm
|
|
lit.llvm.initialize(lit_config, config)
|
|
|
|
# Let the main config do the real work.
|
|
lit_config.load_config(
|
|
config, os.path.join(path(r"@MLIR_SOURCE_DIR@"), "test/lit.cfg.py"))
|