From 5fce9047a00bd627ec9ad03b05986ab1a24a31be Mon Sep 17 00:00:00 2001
From: Hanhan Wang <hanchung@google.com>
Date: Mon, 28 Aug 2023 15:22:12 -0700
Subject: [PATCH] fix lit tests

---
 .../test/transformation_pipeline.mlir         | 21 +++++++------------
 1 file changed, 7 insertions(+), 14 deletions(-)

diff --git a/compiler/src/iree/compiler/GlobalOptimization/test/transformation_pipeline.mlir b/compiler/src/iree/compiler/GlobalOptimization/test/transformation_pipeline.mlir
index 7fedb39c67d9..0482ddf92f84 100644
--- a/compiler/src/iree/compiler/GlobalOptimization/test/transformation_pipeline.mlir
+++ b/compiler/src/iree/compiler/GlobalOptimization/test/transformation_pipeline.mlir
@@ -15,17 +15,10 @@ func.func @elementwiseOps(%arg0 : tensor<4xf32>) -> tensor<4xf32> {
   return %2 : tensor<4xf32>
 }
 
-// CHECK-LABEL: flow.executable private @elementwiseOps_dispatch_0 {
-//  CHECK-NEXT:   flow.executable.export public @elementwiseOps_dispatch_0{{.*}} workgroups() -> (index, index, index) {
-//       CHECK:     %x, %y, %z = flow.dispatch.workgroup_count_from_slice
-//       CHECK:     flow.return %x, %y, %z
-//       CHECK:   module {
-//  CHECK-NEXT:     func.func @elementwiseOps_dispatch_0{{.*}}(%arg0: !flow.dispatch.tensor<readonly:tensor<4xf32>>, %arg1: !flow.dispatch.tensor<writeonly:tensor<4xf32>>) {
-//       CHECK:       %{{.+}} = linalg.generic
-//       CHECK:         %{{.+}} = arith.addf %{{.+}}, %{{.+}} : f32
-//  CHECK-NEXT:         %{{.+}} = arith.subf %{{.+}}, %{{.+}} : f32
-//  CHECK-NEXT:         %{{.+}} = arith.mulf %{{.+}}, %{{.+}} : f32
-//       CHECK: func.func @elementwiseOps(%arg0: tensor<4xf32>) -> tensor<4xf32> {
-//  CHECK-NEXT:   %0 = flow.dispatch @elementwiseOps_dispatch_0::@elementwiseOps_dispatch_0{{.*}}(%arg0) : (tensor<4xf32>) -> tensor<4xf32>
-//  CHECK-NEXT:   return %0 : tensor<4xf32>
-//  CHECK-NEXT: }
+// CHECK-LABEL: func.func @elementwiseOps(%arg0: tensor<4xf32>) -> tensor<4xf32> {
+//       CHECK:   %{{.+}} = linalg.generic
+//       CHECK:     %{{.+}} = arith.addf %{{.+}}, %{{.+}} : f32
+//       CHECK:   %{{.+}} = linalg.generic
+//       CHECK:     %{{.+}} = arith.subf %{{.+}}, %{{.+}} : f32
+//       CHECK:   %{{.+}} = linalg.generic
+//       CHECK:     %{{.+}} = arith.mulf %{{.+}}, %{{.+}} : f32