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>, %arg1: !flow.dispatch.tensor>) { -// 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