| // RUN: [[ $IREE_VULKAN_DISABLE == 1 ]] || (iree-run-mlir -iree-hal-target-backends=vulkan-spirv -iree-use-linalg-to-spirv-path -input-value="4x8xi32=[[1 2 3 4 5 6 7 8][9 10 11 12 13 14 15 16][17 18 19 20 21 22 23 24][25 26 27 28 29 30 31 32]]" -input-value="4x8xi32=[[2 4 6 8 10 12 14 16][18 20 22 24 26 28 30 32][34 36 38 40 42 44 46 48][50 52 54 56 58 60 62 64]]" -input-value="4x8xi32=[[3 6 9 12 15 18 21 24][27 30 33 36 39 42 45 48][51 54 57 60 63 66 69 72][75 78 81 84 87 90 93 96]]" %s | IreeFileCheck %s) |
| |
| // CHECK-LABEL: EXEC @pw_add_mul |
| // CHECK: 4x8xi32=[5 14 27 44 65 90 119 152][189 230 275 324 377 434 495 560][629 702 779 860 945 1034 1127 1224][1325 1430 1539 1652 1769 1890 2015 2144] |
| module { |
| func @pw_add_mul(%arg0: tensor<4x8xi32>, %arg1: tensor<4x8xi32>, %arg2 : tensor<4x8xi32>) -> tensor<4x8xi32> { |
| %0 = "xla_hlo.mul"(%arg0, %arg1) : (tensor<4x8xi32>, tensor<4x8xi32>) -> tensor<4x8xi32> |
| %1 = "xla_hlo.add"(%0, %arg2) : (tensor<4x8xi32>, tensor<4x8xi32>) -> tensor<4x8xi32> |
| return %1 : tensor<4x8xi32> |
| } |
| } |
