bufferization pipeline fixes

cruise-automation · Jun 21, 2024 · 58cf5da · 58cf5da
1 parent a9ae356
commit 58cf5da
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/lib/Pipeline/Pipeline.cpp b/lib/Pipeline/Pipeline.cpp
@@ -78,14 +78,14 @@ static void createTcpToLlvmPipeline(OpPassManager &pm) {
   bufferizationOptions.bufferizeFunctionBoundaries = true;
   bufferizationOptions.setFunctionBoundaryTypeConversion(
           bufferization::LayoutMapOption::IdentityLayoutMap);
-  pm.addNestedPass<func::FuncOp>(bufferization::createOneShotBufferizePass(bufferizationOptions));
+  pm.addPass(bufferization::createOneShotBufferizePass(bufferizationOptions));
   pm.addNestedPass<func::FuncOp>(
       bufferization::createFinalizingBufferizePass());
-  pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());
-  pm.addNestedPass<func::FuncOp>(bufferization::createOwnershipBasedBufferDeallocationPass());
-  pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());
-  pm.addNestedPass<func::FuncOp>(bufferization::createBufferDeallocationSimplificationPass());
-  pm.addNestedPass<func::FuncOp>(bufferization::createLowerDeallocationsPass());
+  pm.addPass(createCanonicalizerPass());
+  pm.addPass(bufferization::createOwnershipBasedBufferDeallocationPass());
+  pm.addPass(createCanonicalizerPass());
+  pm.addPass(bufferization::createBufferDeallocationSimplificationPass());
+  pm.addPass(bufferization::createLowerDeallocationsPass());
   pm.addPass(createCSEPass());
   pm.addPass(createCanonicalizerPass());
   pm.addPass(createBufferizationToMemRefPass());