[Mlir-commits] [mlir] [mlir][EmitC] Update WrapFunInClassPass pass (PR #203641)

[Bhavesh M]

================
@@ -30,34 +32,46 @@ struct WrapFuncInClassPass
   void runOnOperation() override {
     mlir::ModuleOp moduleOp = getOperation();
 
-    llvm::SmallVector<emitc::GlobalOp> globalsToMove;
-    moduleOp.walk(
-        [&](mlir::emitc::GlobalOp op) { globalsToMove.push_back(op); });
+    std::map<FuncOp, llvm::DenseSet<GlobalOp>> globalsUsedByFuncs;
+
+    SymbolTableCollection symbolTable;
+    moduleOp.walk([&globalsUsedByFuncs, &symbolTable](FuncOp funcOp) {
+      funcOp.walk(
+          [&globalsUsedByFuncs, &symbolTable, funcOp](GetGlobalOp getGlobalOp) {
----------------
beamandala wrote:

No, thank you for catching that.

https://github.com/llvm/llvm-project/pull/203641