| Ben Vanik | d5a9b1b | 2023-03-06 12:06:27 -0800 | [diff] [blame^] | 1 | // RUN: iree-compile --compile-to=input %s | FileCheck %s --check-prefix=INPUT-PHASE |
| Ben Vanik | 7efcd0a | 2022-11-29 09:21:01 -0800 | [diff] [blame] | 2 | // INPUT-PHASE: func.func @abs(%[[ARG0:.+]]: tensor<f32>) |
| 3 | // INPUT-PHASE: math.absf %[[ARG0]] : tensor<f32> |
| 4 | |
| Ben Vanik | d5a9b1b | 2023-03-06 12:06:27 -0800 | [diff] [blame^] | 5 | // RUN: iree-compile --compile-to=abi %s | FileCheck %s --check-prefix=ABI-PHASE |
| Ben Vanik | 7efcd0a | 2022-11-29 09:21:01 -0800 | [diff] [blame] | 6 | // ABI-PHASE: func.func @abs(%[[ARG0:.+]]: !hal.buffer_view) |
| 7 | // ABI-PHASE: %[[INPUT:.+]] = hal.tensor.import %[[ARG0]] : !hal.buffer_view -> tensor<f32> |
| 8 | // ABI-PHASE: math.absf %[[INPUT]] : tensor<f32> |
| 9 | |
| Ben Vanik | d5a9b1b | 2023-03-06 12:06:27 -0800 | [diff] [blame^] | 10 | // RUN: iree-compile --compile-to=flow %s | FileCheck %s --check-prefix=FLOW-PHASE |
| Ben Vanik | 7efcd0a | 2022-11-29 09:21:01 -0800 | [diff] [blame] | 11 | // FLOW-PHASE: flow.executable.export public @abs_dispatch_0 |
| 12 | // FLOW-PHASE: flow.dispatch @abs_dispatch_0 |
| 13 | |
| Ben Vanik | d5a9b1b | 2023-03-06 12:06:27 -0800 | [diff] [blame^] | 14 | // RUN: iree-compile --compile-to=stream %s | FileCheck %s --check-prefix=STREAM-PHASE |
| Ben Vanik | 7efcd0a | 2022-11-29 09:21:01 -0800 | [diff] [blame] | 15 | // STREAM-PHASE: stream.executable.export public @abs_dispatch_0 |
| 16 | // STREAM-PHASE: stream.cmd.dispatch @abs_dispatch_0 |
| 17 | |
| Ben Vanik | d5a9b1b | 2023-03-06 12:06:27 -0800 | [diff] [blame^] | 18 | // RUN: iree-compile --compile-to=executable-sources --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=EXECUTABLE-SOURCES-PHASE |
| 19 | // EXECUTABLE-SOURCES-PHASE: hal.executable private @abs_dispatch_0 |
| 20 | // EXECUTABLE-SOURCES-PHASE: hal.executable.variant |
| 21 | // EXECUTABLE-SOURCES-PHASE: linalg.generic |
| 22 | // EXECUTABLE-SOURCES-PHASE: math.absf |
| 23 | |
| 24 | // RUN: iree-compile --compile-to=executable-targets --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=EXECUTABLE-TARGETS-PHASE |
| 25 | // EXECUTABLE-TARGETS-PHASE: hal.executable private @abs_dispatch_0 |
| 26 | // EXECUTABLE-TARGETS-PHASE: hal.executable.variant |
| 27 | // EXECUTABLE-TARGETS-PHASE: vm.abs.f32 |
| 28 | |
| Ben Vanik | 7efcd0a | 2022-11-29 09:21:01 -0800 | [diff] [blame] | 29 | // RUN: iree-compile --compile-to=hal --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=HAL-PHASE |
| 30 | // HAL-PHASE: hal.executable private @abs_dispatch_0 |
| 31 | // HAL-PHASE: hal.executable.binary |
| 32 | // HAL-PHASE: hal.command_buffer.dispatch |
| 33 | |
| 34 | // RUN: iree-compile --compile-to=vm --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=VM-PHASE |
| 35 | // VM-PHASE: vm.rodata private @abs_dispatch_0 |
| 36 | // VM-PHASE: vm.call @hal.command_buffer.dispatch |
| 37 | |
| 38 | // RUN: iree-compile --output-format=vm-asm --compile-to=end --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=END-PHASE |
| 39 | // RUN: iree-compile --output-format=vm-asm --iree-hal-target-backends=vmvx %s | FileCheck %s --check-prefix=END-PHASE |
| 40 | // END-PHASE: vm.rodata private @abs_dispatch_0 |
| 41 | // END-PHASE: vm.call @hal.command_buffer.dispatch |
| 42 | |
| 43 | func.func @abs(%input : tensor<f32>) -> (tensor<f32>) { |
| 44 | %result = math.absf %input : tensor<f32> |
| 45 | return %result : tensor<f32> |
| 46 | } |
