MLIR-AIE
Public Member Functions | List of all members
AIEDMATasksToNPUPass Struct Reference
+ Inheritance diagram for AIEDMATasksToNPUPass:
+ Collaboration diagram for AIEDMATasksToNPUPass:

Public Member Functions

bool shouldSkipBlock (Block &block)
 
LogicalResult verifyBdInBlock (Block &block)
 
LogicalResult verifyOptionalLocksInBlock (Block &block)
 
LogicalResult verifyNoUnsupportedOpsInBlock (Block &block)
 
AIE::DMABDOp getBdForBlock (Block &block)
 
std::optional< std::pair< AIE::UseLockOp, AIE::UseLockOp > > getOptionalLockOpsForBlock (Block &block)
 
LogicalResult setAddressForSingleBD (OpBuilder &builder, AIE::DMABDOp &bd_op, AIE::TileOp &tile)
 
LogicalResult rewriteSingleBD (OpBuilder &builder, Block &block, AIE::TileOp &tile, AIE::DMAChannelDir channelDir)
 
LogicalResult hoistNextBdOpsIntoAttrs (DMAConfigureTaskOp op)
 
LogicalResult rewriteSingleDMAConfigureTaskOp (DMAConfigureTaskOp op)
 
LogicalResult rewriteDMAConfigureTaskOp (AIE::DeviceOp device)
 
void runOnOperation () override
 

Detailed Description

Definition at line 80 of file AIEDMATasksToNPU.cpp.

Member Function Documentation

◆ getBdForBlock()

AIE::DMABDOp AIEDMATasksToNPUPass::getBdForBlock ( Block &  block)
inline

Definition at line 166 of file AIEDMATasksToNPU.cpp.

Referenced by hoistNextBdOpsIntoAttrs(), and rewriteSingleBD().

◆ getOptionalLockOpsForBlock()

std::optional< std::pair< AIE::UseLockOp, AIE::UseLockOp > > AIEDMATasksToNPUPass::getOptionalLockOpsForBlock ( Block &  block)
inline

Definition at line 174 of file AIEDMATasksToNPU.cpp.

◆ hoistNextBdOpsIntoAttrs()

LogicalResult AIEDMATasksToNPUPass::hoistNextBdOpsIntoAttrs ( DMAConfigureTaskOp  op)
inline

Definition at line 415 of file AIEDMATasksToNPU.cpp.

References getBdForBlock(), and shouldSkipBlock().

Referenced by rewriteSingleDMAConfigureTaskOp().

◆ rewriteDMAConfigureTaskOp()

LogicalResult AIEDMATasksToNPUPass::rewriteDMAConfigureTaskOp ( AIE::DeviceOp  device)
inline

Definition at line 504 of file AIEDMATasksToNPU.cpp.

References rewriteSingleDMAConfigureTaskOp().

Referenced by runOnOperation().

◆ rewriteSingleBD()

LogicalResult AIEDMATasksToNPUPass::rewriteSingleBD ( OpBuilder &  builder,
Block &  block,
AIE::TileOp &  tile,
AIE::DMAChannelDir  channelDir 
)
inline

◆ rewriteSingleDMAConfigureTaskOp()

LogicalResult AIEDMATasksToNPUPass::rewriteSingleDMAConfigureTaskOp ( DMAConfigureTaskOp  op)
inline

◆ runOnOperation()

void AIEDMATasksToNPUPass::runOnOperation ( )
inlineoverride

Definition at line 517 of file AIEDMATasksToNPU.cpp.

References rewriteDMAConfigureTaskOp().

◆ setAddressForSingleBD()

LogicalResult AIEDMATasksToNPUPass::setAddressForSingleBD ( OpBuilder &  builder,
AIE::DMABDOp &  bd_op,
AIE::TileOp &  tile 
)
inline

◆ shouldSkipBlock()

bool AIEDMATasksToNPUPass::shouldSkipBlock ( Block &  block)
inline

◆ verifyBdInBlock()

LogicalResult AIEDMATasksToNPUPass::verifyBdInBlock ( Block &  block)
inline

Definition at line 90 of file AIEDMATasksToNPU.cpp.

Referenced by rewriteSingleDMAConfigureTaskOp().

◆ verifyNoUnsupportedOpsInBlock()

LogicalResult AIEDMATasksToNPUPass::verifyNoUnsupportedOpsInBlock ( Block &  block)
inline

Definition at line 140 of file AIEDMATasksToNPU.cpp.

Referenced by rewriteSingleDMAConfigureTaskOp().

◆ verifyOptionalLocksInBlock()

LogicalResult AIEDMATasksToNPUPass::verifyOptionalLocksInBlock ( Block &  block)
inline

Definition at line 126 of file AIEDMATasksToNPU.cpp.

Referenced by rewriteSingleDMAConfigureTaskOp().


The documentation for this struct was generated from the following file: