Speedup RewriteStackPtr pass (#3878)

Signed-off-by: Anatoly Myachev <[email protected]>

Author: anmyachev

third_party/intel/lib/TritonIntelGPUTransforms/RewriteStackPtr.cpp

@@ -25,14 +25,14 @@ struct TritonIntelGPURewriteStackPtrPass
   void runOnOperation() override {
     ModuleOp mod = getOperation();
     MLIRContext *ctx = &getContext();
-    ModuleAllocation allocation(mod);
     LLVM::LLVMPointerType ptrTy =
         ptr_ty(ctx, mlir::triton::TritonGEN::TritonGENMemorySpace::kWorkgroup);
 
     auto globalSmem = mod.lookupSymbol<LLVM::GlobalOp>("global_smem");
     if (!globalSmem) {
       return;
     }
+    CallGraph<Allocation> allocation(mod);
     bool usePoison =
         (mod->getAttrOfType<IntegerAttr>("ttg.shared").getInt() == 0);