[NFC] nest passes better

llvm · May 21, 2024 · 535f147 · 535f147
1 parent fd491e1
commit 535f147
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/lib/Firtool/Firtool.cpp b/lib/Firtool/Firtool.cpp
@@ -119,10 +119,8 @@ LogicalResult firtool::populateCHIRRTLToLowFIRRTL(mlir::PassManager &pm,
   pm.addNestedPass<firrtl::CircuitOp>(firrtl::createLowerFIRRTLTypesPass(
       opt.getPreserveAggregate(), firrtl::PreserveAggregate::None));
 
-  pm.nest<firrtl::CircuitOp>().nestAny().addPass(
-      firrtl::createExpandWhensPass());
-
   auto &modulePM = pm.nest<firrtl::CircuitOp>().nest<firrtl::FModuleOp>();
+  modulePM.addPass(firrtl::createExpandWhensPass());
   modulePM.addPass(firrtl::createSFCCompatPass());
   modulePM.addPass(firrtl::createLayerMergePass());
   modulePM.addPass(firrtl::createLayerSinkPass());