| Ben Vanik | 9461d3b | 2023-04-18 16:39:25 -0700 | [diff] [blame] | 1 | // RUN: iree-compile --iree-hal-target-backends=vmvx %s | iree-benchmark-module --device=local-task --module=- | FileCheck %s |
| 2 | // RUN: [[ $IREE_VULKAN_DISABLE == 1 ]] || (iree-compile --iree-hal-target-backends=vulkan-spirv %s | iree-benchmark-module --device=vulkan --module=- | FileCheck %s) |
| Han-Chung Wang | 00aa2fc | 2020-10-12 02:05:39 -0700 | [diff] [blame] | 3 | |
| 4 | module { |
| Thomas | f0f64ca | 2022-03-30 10:38:01 -0700 | [diff] [blame] | 5 | func.func @foo1() -> tensor<4xf32> { |
| Ben Vanik | a1ff64e | 2021-08-06 10:13:35 -0700 | [diff] [blame] | 6 | %input = util.unfoldable_constant dense<[0.0, 1.0, 2.0, 4.0]> : tensor<4xf32> |
| Scott Todd | 7e46380 | 2022-06-30 14:31:00 -0700 | [diff] [blame] | 7 | %result = math.exp %input : tensor<4xf32> |
| Han-Chung Wang | 00aa2fc | 2020-10-12 02:05:39 -0700 | [diff] [blame] | 8 | return %result : tensor<4xf32> |
| 9 | } |
| Thomas | f0f64ca | 2022-03-30 10:38:01 -0700 | [diff] [blame] | 10 | func.func @foo2() -> tensor<4xf32> { |
| Ben Vanik | a1ff64e | 2021-08-06 10:13:35 -0700 | [diff] [blame] | 11 | %input = util.unfoldable_constant dense<[0.0, 1.0, 2.0, 4.0]> : tensor<4xf32> |
| Thomas | 4c2029f | 2022-08-15 19:20:14 -0700 | [diff] [blame] | 12 | %result = math.absf %input : tensor<4xf32> |
| Han-Chung Wang | 00aa2fc | 2020-10-12 02:05:39 -0700 | [diff] [blame] | 13 | return %result : tensor<4xf32> |
| 14 | } |
| 15 | } |
| 16 | // CHECK: BM_foo1 |
| 17 | // CHECK: BM_foo2 |
