| from lit.llvm import llvm_config |
| |
| if not config.mlir_include_integration_tests: |
| config.unsupported = True |
| |
| |
| def configure_aarch64_lli_and_mcr_cmd(): |
| lli_cmd = "lli" |
| mcr_cmd = "mlir-cpu-runner" |
| |
| # NOTE: If the SVE tests are disabled and the SME tests are enabled to run |
| # under emulation, the SVE specific RUN lines in the SparseTensor tests |
| # will run under emulation. |
| if not (config.mlir_run_arm_sve_tests or config.mlir_run_arm_sme_tests): |
| return (lli_cmd, mcr_cmd) |
| |
| config.substitutions.append( |
| ( |
| "%mlir_native_utils_lib_dir", |
| config.arm_emulator_utils_lib_dir or config.mlir_lib_dir, |
| ) |
| ) |
| |
| if config.arm_emulator_executable: |
| if config.arm_emulator_lli_executable: |
| lli_cmd = config.arm_emulator_lli_executable |
| else: |
| # Top-level lit config adds llvm_tools_dir to PATH but this is lost |
| # when running under an emulator. If the user didn't specify an lli |
| # executable, use absolute path %llvm_tools_dir/lli. |
| lli_cmd = llvm_config.use_llvm_tool( |
| "lli", |
| search_env="LLI", |
| required=True, |
| search_paths=[config.llvm_tools_dir], |
| use_installed=False, |
| ) |
| |
| if config.arm_emulator_mlir_cpu_runner_executable: |
| mcr_cmd = config.arm_emulator_mlir_cpu_runner_executable |
| else: |
| # Top-level LIT config adds llvm_tools_dir to PATH but this is lost |
| # when running under an emulator. If the user didn't specify an |
| # mlir-cpu-runner executable, use absolute path |
| # %llvm_tools_dir/mlir-cpu-runner. |
| mcr_cmd = llvm_config.use_llvm_tool( |
| "mlir-cpu-runner", |
| search_env="MLIR_CPU_RUNNER", |
| required=True, |
| search_paths=[config.mlir_tools_dir], |
| use_installed=False, |
| ) |
| |
| # Run test in emulator (qemu or armie) |
| emulation_cmd = ( |
| f"{config.arm_emulator_executable} {config.arm_emulator_options}" |
| ) |
| |
| lli_cmd = f"{emulation_cmd} {lli_cmd}" |
| mcr_cmd = f"{emulation_cmd} {mcr_cmd}" |
| |
| return (lli_cmd, mcr_cmd) |
| |
| |
| aarch64_lli_cmd, aarch64_mcr_cmd = configure_aarch64_lli_and_mcr_cmd() |
| |
| # Configure the following AArch64 substitutions: |
| # |
| # * %lli_aarch64_cmd - Invokes lli. For tests that _will_ run on AArch64 (ArmSVE, ArmSME). |
| # * %lli_host_or_aarch64_cmd - Invokes lli. For tests that _may_ run on AArch64 (SparseTensor). |
| # * %mcr_aarch64_cmd - Invokes mlir-cpu-runner. For tests that _will_ |
| # run on AArch64. May invoke mlir-cpu-runner under |
| # an AArch64 emulator (when |
| # `config.arm_emulator_executable` is set). |
| # |
| # AArch64 tests will run under emulation if configured at build time by the |
| # following CMake options: |
| # |
| # * ARM_EMULATOR_EXECUTABLE - emulator to use. |
| # * ARM_EMULATOR_OPTIONS - options for emulator. |
| # * ARM_EMULATOR_LLI_EXECUTABLE - AArch64 native lli to support cross-compilation. |
| # * ARM_EMULATOR_UTILS_LIB_DIR - AArch64 native utilites library to support cross-compilation. |
| # |
| # Functionally the two substitutions are equivalent, i.e. %lli_aarch64_cmd |
| # could be used in the SparseTensor tests where necessary, but the meaning |
| # conveyed by the substitution name would be a misnomer if the host target |
| # is not AArch64 and MLIR_RUN_ARM_SVE_TESTS=OFF. |
| config.substitutions.append(("%lli_aarch64_cmd", aarch64_lli_cmd)) |
| config.substitutions.append(("%lli_host_or_aarch64_cmd", aarch64_lli_cmd)) |
| config.substitutions.append(("%mcr_aarch64_cmd", aarch64_mcr_cmd)) |