diff --git a/mlir/lib/Dialect/Bufferization/Transforms/LowerDeallocations.cpp b/mlir/lib/Dialect/Bufferization/Transforms/LowerDeallocations.cpp index c7052434f2c3b..982d9558d3132 100644 --- a/mlir/lib/Dialect/Bufferization/Transforms/LowerDeallocations.cpp +++ b/mlir/lib/Dialect/Bufferization/Transforms/LowerDeallocations.cpp @@ -432,6 +432,7 @@ func::FuncOp mlir::bufferization::buildDeallocationLibraryFunction( // Generate the func operation itself. auto helperFuncOp = func::FuncOp::create( loc, "dealloc_helper", builder.getFunctionType(argTypes, {})); + helperFuncOp.setVisibility(SymbolTable::Visibility::Private); symbolTable.insert(helperFuncOp); auto &block = helperFuncOp.getFunctionBody().emplaceBlock(); block.addArguments(argTypes, SmallVector(argTypes.size(), loc)); diff --git a/mlir/test/Dialect/Bufferization/Transforms/lower-deallocations.mlir b/mlir/test/Dialect/Bufferization/Transforms/lower-deallocations.mlir index 19d3bbf7089c1..2c69fcab08a8d 100644 --- a/mlir/test/Dialect/Bufferization/Transforms/lower-deallocations.mlir +++ b/mlir/test/Dialect/Bufferization/Transforms/lower-deallocations.mlir @@ -120,7 +120,7 @@ func.func @conversion_dealloc_multiple_memrefs_and_retained(%arg0: memref<2xf32> // CHECK: memref.dealloc [[RETAIN_CONDS]] // CHECK: return [[OWNERSHIP0]], [[OWNERSHIP1]] -// CHECK: func @dealloc_helper +// CHECK: func private @dealloc_helper // CHECK-SAME: ([[TO_DEALLOC_MR:%.+]]: memref, [[TO_RETAIN_MR:%.+]]: memref, // CHECK-SAME: [[CONDS:%.+]]: memref, [[DEALLOC_CONDS_OUT:%.+]]: memref, // CHECK-SAME: [[RETAIN_CONDS_OUT:%.+]]: memref)