fix lit tests

iree-org · Aug 28, 2023 · 5fce904 · 5fce904
1 parent 097ad2a
commit 5fce904
Showing 1 changed file with 7 additions and 14 deletions.
diff --git 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