Fixing the --compile-to= flag and adding a test of all phases. (#11345)
Fixes #11330.
diff --git a/tools/test/compile_to_phase.mlir b/tools/test/compile_to_phase.mlir
new file mode 100644
index 0000000..cb3a8ed
--- /dev/null
+++ b/tools/test/compile_to_phase.mlir
@@ -0,0 +1,35 @@
+// RUN: iree-compile --compile-to=input --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=INPUT-PHASE
+// INPUT-PHASE: func.func @abs(%[[ARG0:.+]]: tensor<f32>)
+// INPUT-PHASE: math.absf %[[ARG0]] : tensor<f32>
+
+// RUN: iree-compile --compile-to=abi --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=ABI-PHASE
+// ABI-PHASE: func.func @abs(%[[ARG0:.+]]: !hal.buffer_view)
+// ABI-PHASE: %[[INPUT:.+]] = hal.tensor.import %[[ARG0]] : !hal.buffer_view -> tensor<f32>
+// ABI-PHASE: math.absf %[[INPUT]] : tensor<f32>
+
+// RUN: iree-compile --compile-to=flow --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=FLOW-PHASE
+// FLOW-PHASE: flow.executable.export public @abs_dispatch_0
+// FLOW-PHASE: flow.dispatch @abs_dispatch_0
+
+// RUN: iree-compile --compile-to=stream --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=STREAM-PHASE
+// STREAM-PHASE: stream.executable.export public @abs_dispatch_0
+// STREAM-PHASE: stream.cmd.dispatch @abs_dispatch_0
+
+// RUN: iree-compile --compile-to=hal --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=HAL-PHASE
+// HAL-PHASE: hal.executable private @abs_dispatch_0
+// HAL-PHASE: hal.executable.binary
+// HAL-PHASE: hal.command_buffer.dispatch
+
+// RUN: iree-compile --compile-to=vm --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=VM-PHASE
+// VM-PHASE: vm.rodata private @abs_dispatch_0
+// VM-PHASE: vm.call @hal.command_buffer.dispatch
+
+// RUN: iree-compile --output-format=vm-asm --compile-to=end --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=END-PHASE
+// RUN: iree-compile --output-format=vm-asm --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=END-PHASE
+// END-PHASE: vm.rodata private @abs_dispatch_0
+// END-PHASE: vm.call @hal.command_buffer.dispatch
+
+func.func @abs(%input : tensor<f32>) -> (tensor<f32>) {
+ %result = math.absf %input : tensor<f32>
+ return %result : tensor<f32>
+}
