Skip to content

Commit c78cf3f

Browse files
authored
Enable AIROptimizeShimDMABDsOptions when registering AIROptimizeShimDMABDs pass (Xilinx#852)
1 parent bf6d10f commit c78cf3f

File tree

2 files changed

+9
-0
lines changed

2 files changed

+9
-0
lines changed

mlir/include/air/Transform/AIRDependencyScheduleOpt.h

+2
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,8 @@ std::unique_ptr<mlir::Pass> createAIRIsolateAsyncDmaLoopNests();
6060
std::unique_ptr<mlir::Pass> createAIRLoopFusion();
6161

6262
std::unique_ptr<mlir::Pass> createAIROptimizeShimDMABDs();
63+
std::unique_ptr<Pass>
64+
createAIROptimizeShimDMABDs(AIROptimizeShimDMABDsOptions options);
6365

6466
std::unique_ptr<mlir::Pass> createAIRFuseAllocDealloc();
6567

mlir/lib/Transform/AIRDependencyScheduleOpt.cpp

+7
Original file line numberDiff line numberDiff line change
@@ -5600,6 +5600,9 @@ class AIROptimizeShimDMABDs
56005600
public:
56015601
AIROptimizeShimDMABDs() = default;
56025602
AIROptimizeShimDMABDs(const AIROptimizeShimDMABDs &pass) {}
5603+
AIROptimizeShimDMABDs(
5604+
const ::xilinx::air::AIROptimizeShimDMABDsOptions &options)
5605+
: AIROptimizeShimDMABDsBase(options) {}
56035606

56045607
void getDependentDialects(::mlir::DialectRegistry &registry) const override {
56055608
registry.insert<scf::SCFDialect, air::airDialect, AIE::AIEDialect>();
@@ -6108,6 +6111,10 @@ std::unique_ptr<Pass> createAIRLoopFusion() {
61086111
std::unique_ptr<Pass> createAIROptimizeShimDMABDs() {
61096112
return std::make_unique<AIROptimizeShimDMABDs>();
61106113
}
6114+
std::unique_ptr<Pass>
6115+
createAIROptimizeShimDMABDs(AIROptimizeShimDMABDsOptions options) {
6116+
return std::make_unique<AIROptimizeShimDMABDs>(options);
6117+
}
61116118

61126119
std::unique_ptr<Pass> createAIRFuseAllocDealloc() {
61136120
return std::make_unique<AIRFuseAllocDealloc>();

0 commit comments

Comments
 (0)