diff --git a/mlir/test/Dialect/Linalg/detensorize_0d.mlir b/mlir/test/Dialect/Linalg/detensorize_0d.mlir index 13baab32e49ea..74931cb0830bc 100644 --- a/mlir/test/Dialect/Linalg/detensorize_0d.mlir +++ b/mlir/test/Dialect/Linalg/detensorize_0d.mlir @@ -2,7 +2,7 @@ #map = affine_map<() -> ()> -func.func @detensor_simple(%arg1: tensor, %arg2: tensor) -> tensor attributes {iree.module.export} { +func.func @detensor_simple(%arg1: tensor, %arg2: tensor) -> tensor { %0 = tensor.empty() : tensor %1 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = []} ins(%arg1, %arg2 : tensor, tensor) @@ -21,7 +21,7 @@ func.func @detensor_simple(%arg1: tensor, %arg2: tensor) -> tensor, %arg2: tensor) -> tensor attributes {iree.module.export} { +func.func @detensor_op_sequence(%arg1: tensor, %arg2: tensor) -> tensor { %0 = tensor.empty() : tensor %1 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = []} ins(%arg1, %arg2 : tensor, tensor) @@ -61,7 +61,7 @@ func.func @detensor_op_sequence(%arg1: tensor, %arg2: tensor) -> tenso // CHECK: %[[new_tensor_res:.*]] = tensor.from_elements %[[detensored_res3]] // CHECK: return %[[new_tensor_res]] -func.func @detensor_multiple_ops(%arg1: tensor, %arg2: tensor) -> tensor attributes {iree.module.export} { +func.func @detensor_multiple_ops(%arg1: tensor, %arg2: tensor) -> tensor { %0 = tensor.empty() : tensor %1 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = []} ins(%arg1, %arg2 : tensor, tensor) @@ -82,7 +82,7 @@ func.func @detensor_multiple_ops(%arg1: tensor, %arg2: tensor) -> tens // CHECK: %[[new_tensor_res:.*]] = tensor.from_elements %[[detensored_res2]] // CHECK: return %[[new_tensor_res]] -func.func @detensor_foreign_op(%arg1: tensor, %arg2: tensor) -> tensor attributes {iree.module.export} { +func.func @detensor_foreign_op(%arg1: tensor, %arg2: tensor) -> tensor { %0 = tensor.empty() : tensor %1 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = []} ins(%arg1, %arg2 : tensor, tensor)