| @LIT_SITE_CFG_IN_HEADER@ |
| |
| import sys |
| |
| config.target_triple = "@LLVM_TARGET_TRIPLE@" |
| config.llvm_src_root = "@LLVM_SOURCE_DIR@" |
| config.llvm_tools_dir = lit_config.substitute("@LLVM_TOOLS_DIR@") |
| 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.native_target = "@LLVM_NATIVE_ARCH@" |
| config.host_os = "@HOST_OS@" |
| config.host_cc = "@HOST_CC@" |
| config.host_cxx = "@HOST_CXX@" |
| config.enable_libcxx = "@LLVM_ENABLE_LIBCXX@" |
| config.host_cmake = "@CMAKE_COMMAND@" |
| config.host_cmake_generator = "@CMAKE_GENERATOR@" |
| config.llvm_use_linker = "@LLVM_USE_LINKER@" |
| config.llvm_use_sanitizer = "@LLVM_USE_SANITIZER@" |
| config.host_arch = "@HOST_ARCH@" |
| config.mlir_src_root = "@MLIR_SOURCE_DIR@" |
| config.mlir_obj_root = "@MLIR_BINARY_DIR@" |
| config.mlir_tools_dir = "@MLIR_TOOLS_DIR@" |
| config.mlir_cmake_dir = "@MLIR_CMAKE_DIR@" |
| config.mlir_lib_dir = "@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 = @MLIR_ENABLE_ROCM_CONVERSIONS@ |
| 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.enable_sycl_runner = @MLIR_ENABLE_SYCL_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.intel_sde_executable = "@INTEL_SDE_EXECUTABLE@" |
| config.mlir_run_amx_tests = @MLIR_RUN_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_x86vector_tests = @MLIR_RUN_X86VECTOR_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.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, "@MLIR_SOURCE_DIR@/test/lit.cfg.py") |