1 files changed, 200 insertions, 0 deletions
diff --git a/backend/tools/MIRTool/MIRToolStageTest.cmake b/backend/tools/MIRTool/MIRToolStageTest.cmake
new file mode 100644
index 0000000..0d3ef2e
--- /dev/null
+++ b/backend/tools/MIRTool/MIRToolStageTest.cmake
@@ -0,0 +1,200 @@
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+if(NOT DEFINED TOOL OR NOT DEFINED INPUT OR NOT DEFINED WORK_DIRECTORY)
+  message(FATAL_ERROR "MIR tool stage test is missing a required path")
+endif()
+
+file(REMOVE_RECURSE "${WORK_DIRECTORY}")
+file(MAKE_DIRECTORY "${WORK_DIRECTORY}")
+
+set(
+  Stages
+  prelegalizer-mir
+  legalized-mir
+  regbank-mir
+  selected-mir
+  post-ssa-mir
+  post-ra-mir
+  final-mir
+)
+set(
+  StageOptions
+  --verify
+  -stop-after=legalizer
+  -stop-after=regbankselect
+  -stop-after=instruction-select
+  -stop-after=twoaddressinstruction
+  -stop-after=PBQPAllocator
+  -stop-after=final-encodability-verifier
+)
+set(StageIndex 0)
+foreach(Stage IN LISTS Stages)
+  list(GET StageOptions ${StageIndex} StageOption)
+  set(StageOutput "${WORK_DIRECTORY}/${Stage}.mir")
+  set(RepeatedOutput "${WORK_DIRECTORY}/${Stage}.repeat.mir")
+  set(CanonicalOutput "${WORK_DIRECTORY}/${Stage}.canonical.mir")
+  execute_process(
+    COMMAND "${TOOL}" "${INPUT}" "${StageOption}" -o "${StageOutput}"
+    RESULT_VARIABLE StageResult
+    ERROR_VARIABLE StageDiagnostics
+  )
+  if(NOT StageResult EQUAL 0)
+    message(FATAL_ERROR "${Stage} production failed: ${StageDiagnostics}")
+  endif()
+  execute_process(
+    COMMAND "${TOOL}" "${INPUT}" "${StageOption}" -o "${RepeatedOutput}"
+    RESULT_VARIABLE RepeatedResult
+    ERROR_VARIABLE RepeatedDiagnostics
+  )
+  if(NOT RepeatedResult EQUAL 0)
+    message(FATAL_ERROR "repeated ${Stage} production failed: ${RepeatedDiagnostics}")
+  endif()
+  execute_process(
+    COMMAND "${CMAKE_COMMAND}" -E compare_files "${StageOutput}" "${RepeatedOutput}"
+    RESULT_VARIABLE DeterminismResult
+  )
+  if(NOT DeterminismResult EQUAL 0)
+    message(FATAL_ERROR "${Stage} output is not deterministic")
+  endif()
+  execute_process(
+    COMMAND "${TOOL}" --verify "${StageOutput}" -o "${CanonicalOutput}"
+    RESULT_VARIABLE VerifyResult
+    ERROR_VARIABLE VerifyDiagnostics
+  )
+  if(NOT VerifyResult EQUAL 0)
+    message(FATAL_ERROR "${Stage} output failed strict MIR reload: ${VerifyDiagnostics}")
+  endif()
+  execute_process(
+    COMMAND "${CMAKE_COMMAND}" -E compare_files "${StageOutput}" "${CanonicalOutput}"
+    RESULT_VARIABLE CanonicalResult
+  )
+  if(NOT CanonicalResult EQUAL 0)
+    message(FATAL_ERROR "${Stage} output is not canonical after strict reload")
+  endif()
+  math(EXPR StageIndex "${StageIndex} + 1")
+endforeach()
+
+set(LLVMContractOutput "${WORK_DIRECTORY}/llvm-contract.mir")
+execute_process(
+  COMMAND
+    "${TOOL}" --verify "${INPUT}" -mtriple=riscv64
+    -mattr=+m,+a,+f,+d,+c -relocation-model=pic -code-model=small -simplify-mir
+    -verify-machineinstrs -o "${LLVMContractOutput}"
+  RESULT_VARIABLE LLVMContractResult
+  ERROR_VARIABLE LLVMContractDiagnostics
+)
+if(NOT LLVMContractResult EQUAL 0)
+  message(FATAL_ERROR "LLVM llc-compatible invocation failed: ${LLVMContractDiagnostics}")
+endif()
+file(SIZE "${LLVMContractOutput}" LLVMContractOutputSize)
+if(LLVMContractOutputSize EQUAL 0)
+  message(FATAL_ERROR "LLVM llc-compatible invocation produced no MIR")
+endif()
+
+foreach(
+  RemovedOption
+  IN ITEMS
+    --features=+m
+    --output=-
+)
+  execute_process(
+    COMMAND "${TOOL}" "${INPUT}" "${RemovedOption}"
+    RESULT_VARIABLE RemovedOptionResult
+    ERROR_VARIABLE RemovedOptionDiagnostics
+  )
+  if(RemovedOptionResult EQUAL 0 OR
+     NOT RemovedOptionDiagnostics MATCHES "unknown command-line option")
+    message(FATAL_ERROR "removed option ${RemovedOption} remains accepted")
+  endif()
+endforeach()
+
+set(PassList "${WORK_DIRECTORY}/passes.txt")
+execute_process(
+  COMMAND "${TOOL}" --list-passes "${INPUT}" -o "${PassList}"
+  RESULT_VARIABLE ListResult
+  ERROR_VARIABLE ListDiagnostics
+)
+if(NOT ListResult EQUAL 0)
+  message(FATAL_ERROR "pass listing failed: ${ListDiagnostics}")
+endif()
+file(READ "${PassList}" PassListText)
+foreach(RequiredPass IN ITEMS legalizer regbankselect instruction-select prologepilog)
+  if(NOT PassListText MATCHES "(^|\n)${RequiredPass}(\n|$)")
+    message(FATAL_ERROR "canonical pass list is missing ${RequiredPass}")
+  endif()
+endforeach()
+
+set(LegalizedOutput "${WORK_DIRECTORY}/run-pass.mir")
+execute_process(
+  COMMAND
+    "${TOOL}" "${INPUT}" -run-pass legalizer -o
+    "${LegalizedOutput}"
+  RESULT_VARIABLE RunPassResult
+  ERROR_VARIABLE RunPassDiagnostics
+)
+if(NOT RunPassResult EQUAL 0)
+  message(FATAL_ERROR "single-pass execution failed: ${RunPassDiagnostics}")
+endif()
+
+set(RegBankOutput "${WORK_DIRECTORY}/passes.mir")
+execute_process(
+  COMMAND
+    "${TOOL}" "${INPUT}" -passes legalizer,regbankselect -o
+    "${RegBankOutput}"
+  RESULT_VARIABLE PassesResult
+  ERROR_VARIABLE PassesDiagnostics
+)
+if(NOT PassesResult EQUAL 0)
+  message(FATAL_ERROR "explicit pass pipeline failed: ${PassesDiagnostics}")
+endif()
+
+set(StoppedOutput "${WORK_DIRECTORY}/stopped.mir")
+execute_process(
+  COMMAND
+    "${TOOL}" "${INPUT}" -stop-after legalizer -o
+    "${StoppedOutput}"
+  RESULT_VARIABLE StopResult
+  ERROR_VARIABLE StopDiagnostics
+)
+if(NOT StopResult EQUAL 0)
+  message(FATAL_ERROR "stop-after control failed: ${StopDiagnostics}")
+endif()
+execute_process(
+  COMMAND "${TOOL}" --verify "${StoppedOutput}"
+  RESULT_VARIABLE StopVerifyResult
+  ERROR_VARIABLE StopVerifyDiagnostics
+)
+if(NOT StopVerifyResult EQUAL 0)
+  message(FATAL_ERROR "stop-after output failed strict reload: ${StopVerifyDiagnostics}")
+endif()
+
+execute_process(
+  COMMAND
+    "${TOOL}" "${INPUT}" -filetype=null -print-before legalizer -print-after legalizer
+  RESULT_VARIABLE PrintResult
+  ERROR_VARIABLE PrintDiagnostics
+)
+if(NOT PrintResult EQUAL 0)
+  message(FATAL_ERROR "pass snapshot controls failed: ${PrintDiagnostics}")
+endif()
+if(NOT PrintDiagnostics MATCHES "[*][*][*] prelegalizer-mir legalizer#1" OR
+   NOT PrintDiagnostics MATCHES "[*][*][*] legalized-mir legalizer#1")
+  message(FATAL_ERROR "before/after pass snapshots are incomplete")
+endif()
+
+execute_process(
+  COMMAND
+    "${TOOL}" "${INPUT}" -filetype=null --stage-snapshots
+  RESULT_VARIABLE SnapshotResult
+  ERROR_VARIABLE SnapshotDiagnostics
+)
+if(NOT SnapshotResult EQUAL 0)
+  message(FATAL_ERROR "stage snapshot execution failed: ${SnapshotDiagnostics}")
+endif()
+foreach(Stage IN LISTS Stages)
+  if(NOT SnapshotDiagnostics MATCHES "[*][*][*] ${Stage}")
+    message(FATAL_ERROR "stage snapshots are missing ${Stage}")
+  endif()
+endforeach()
+
+file(REMOVE_RECURSE "${WORK_DIRECTORY}")