mlir/test/lit.site.cfg.py.in

   1 @LIT_SITE_CFG_IN_HEADER@
   2
   3 import sys
   4
   5 config.target_triple = "@LLVM_TARGET_TRIPLE@"
   6 config.llvm_src_root = "@LLVM_SOURCE_DIR@"
   7 config.llvm_tools_dir = lit_config.substitute("@LLVM_TOOLS_DIR@")
   8 config.llvm_shlib_ext = "@SHLIBEXT@"
   9 config.llvm_shlib_dir = lit_config.substitute(path(r"@SHLIBDIR@"))
  10 config.python_executable = "@Python3_EXECUTABLE@"
  11 config.enable_assertions = @ENABLE_ASSERTIONS@
  12 config.native_target = "@LLVM_NATIVE_ARCH@"
  13 config.host_os = "@HOST_OS@"
  14 config.host_cc = "@HOST_CC@"
  15 config.host_cxx = "@HOST_CXX@"
  16 config.enable_libcxx = "@LLVM_ENABLE_LIBCXX@"
  17 config.host_cmake = "@CMAKE_COMMAND@"
  18 config.host_cmake_generator = "@CMAKE_GENERATOR@"
  19 config.llvm_use_linker = "@LLVM_USE_LINKER@"
  20 config.llvm_use_sanitizer = "@LLVM_USE_SANITIZER@"
  21 config.host_arch = "@HOST_ARCH@"
  22 config.mlir_src_root = "@MLIR_SOURCE_DIR@"
  23 config.mlir_obj_root = "@MLIR_BINARY_DIR@"
  24 config.mlir_tools_dir = "@MLIR_TOOLS_DIR@"
  25 config.mlir_cmake_dir = "@MLIR_CMAKE_DIR@"
  26 config.mlir_lib_dir = "@MLIR_LIB_DIR@"
  27
  28 config.build_examples = @LLVM_BUILD_EXAMPLES@
  29 config.run_nvptx_tests = @LLVM_HAS_NVPTX_TARGET@
  30 config.enable_cuda_runner = @MLIR_ENABLE_CUDA_RUNNER@
  31 config.run_rocm_tests = @MLIR_ENABLE_ROCM_CONVERSIONS@
  32 config.enable_rocm_runner = @MLIR_ENABLE_ROCM_RUNNER@
  33 config.gpu_compilation_format = "@MLIR_GPU_COMPILATION_TEST_FORMAT@"
  34 config.rocm_test_chipset = "@ROCM_TEST_CHIPSET@"
  35 config.enable_sycl_runner = @MLIR_ENABLE_SYCL_RUNNER@
  36 config.enable_spirv_cpu_runner = @MLIR_ENABLE_SPIRV_CPU_RUNNER@
  37 config.enable_vulkan_runner = @MLIR_ENABLE_VULKAN_RUNNER@
  38 config.enable_bindings_python = @MLIR_ENABLE_BINDINGS_PYTHON@
  39 config.intel_sde_executable = "@INTEL_SDE_EXECUTABLE@"
  40 config.mlir_run_amx_tests = @MLIR_RUN_AMX_TESTS@
  41 config.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.
  46 if config.mlir_run_arm_sve_tests:
  47     config.available_features.add("mlir_arm_sve_tests")
  48 config.mlir_run_arm_sme_tests = @MLIR_RUN_ARM_SME_TESTS@
  49 config.mlir_run_x86vector_tests = @MLIR_RUN_X86VECTOR_TESTS@
  50 config.mlir_run_riscv_vector_tests = "@MLIR_RUN_RISCV_VECTOR_TESTS@"
  51 config.mlir_run_cuda_tensor_core_tests = @MLIR_RUN_CUDA_TENSOR_CORE_TESTS@
  52 config.mlir_run_cuda_sm80_tests = @MLIR_RUN_CUDA_SM80_TESTS@
  53 config.mlir_run_cuda_sm80_lt_tests = @MLIR_RUN_CUDA_SM80_LT_TESTS@
  54 config.mlir_run_cuda_sm90_tests = @MLIR_RUN_CUDA_SM90_TESTS@
  55 config.mlir_include_integration_tests = @MLIR_INCLUDE_INTEGRATION_TESTS@
  56 config.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.
  60 if config.arm_emulator_executable:
  61     config.available_features.add("mlir_arm_emulator")
  62 config.arm_emulator_options = "@ARM_EMULATOR_OPTIONS@"
  63 config.arm_emulator_mlir_cpu_runner_executable = "@ARM_EMULATOR_MLIR_CPU_RUNNER_EXECUTABLE@"
  64 config.arm_emulator_lli_executable = "@ARM_EMULATOR_LLI_EXECUTABLE@"
  65 config.arm_emulator_utils_lib_dir = "@ARM_EMULATOR_UTILS_LIB_DIR@"
  66 config.arm_sme_abi_routines_shlib = "@ARM_SME_ABI_ROUTINES_SHLIB@"
  67 config.riscv_vector_emulator_executable = "@RISCV_VECTOR_EMULATOR_EXECUTABLE@"
  68 config.riscv_vector_emulator_options = "@RISCV_VECTOR_EMULATOR_OPTIONS@"
  69 config.riscv_emulator_lli_executable = "@RISCV_EMULATOR_LLI_EXECUTABLE@"
  70 config.riscv_emulator_utils_lib_dir = "@RISCV_EMULATOR_UTILS_LIB_DIR@"
  71
  72 import lit.llvm
  73 lit.llvm.initialize(lit_config, config)
  74
  75 # Let the main config do the real work.
  76 lit_config.load_config(config, "@MLIR_SOURCE_DIR@/test/lit.cfg.py")