ShardingPropagation Struct Reference
Inheritance diagram for ShardingPropagation:
Public Member Functions | |
| void | runOnOperation () override |
| The polymorphic API that runs the pass over the currently held operation. | |
| Public Member Functions inherited from mlir::shard::impl::ShardingPropagationBase< ShardingPropagation > | |
| ShardingPropagationBase () | |
| ShardingPropagationBase & | operator= (const ShardingPropagationBase &)=delete |
| ~ShardingPropagationBase ()=default | |
| ::llvm::StringRef | getArgument () const override |
| Return the command line argument used when registering this pass. | |
| ::llvm::StringRef | getDescription () const override |
| Return the command line description used when registering this pass. | |
| ::llvm::StringRef | getName () const override |
| Returns the derived pass name. | |
| std::unique_ptr<::mlir::Pass > | clonePass () const override |
| A clone method to create a copy of this pass. | |
| void | getDependentDialects (::mlir::DialectRegistry ®istry) const override |
| Return the dialect that must be loaded in the context before this pass. | |
| Public Member Functions inherited from mlir::OperationPass< OpT > | |
| ~OperationPass () override=default | |
| Public Member Functions inherited from mlir::Pass | |
| virtual | ~Pass ()=default |
| TypeID | getTypeID () const |
| Returns the unique identifier that corresponds to this pass. | |
| const PassInfo * | lookupPassInfo () const |
| Returns the pass info for this pass, or null if unknown. | |
| virtual void | getDependentDialects (DialectRegistry ®istry) const |
| Register dependent dialects for the current pass. | |
| std::optional< StringRef > | getOpName () const |
| Returns the name of the operation that this pass operates on, or std::nullopt if this is a generic OperationPass. | |
| virtual LogicalResult | initializeOptions (StringRef options, function_ref< LogicalResult(const Twine &)> errorHandler) |
| Attempt to initialize the options of this pass from the given string. | |
| void | printAsTextualPipeline (raw_ostream &os, bool pretty=false) |
| Prints out the pass in the textual representation of pipelines. | |
| ArrayRef< Statistic * > | getStatistics () const |
| Returns the main statistics for this pass instance. | |
| MutableArrayRef< Statistic * > | getStatistics () |
| const Pass * | getThreadingSibling () const |
| Returns the thread sibling of this pass. | |
| const Pass * | getThreadingSiblingOrThis () const |
| Returns the thread sibling of this pass, or the pass itself it has no sibling. | |
Additional Inherited Members | |
| Public Types inherited from mlir::shard::impl::ShardingPropagationBase< ShardingPropagation > | |
| using | Base |
| Static Public Member Functions inherited from mlir::shard::impl::ShardingPropagationBase< ShardingPropagation > | |
| static constexpr ::llvm::StringLiteral | getArgumentName () |
| Returns the command-line argument attached to this pass. | |
| static constexpr ::llvm::StringLiteral | getPassName () |
| Returns the derived pass name. | |
| static bool | classof (const ::mlir::Pass *pass) |
| Support isa/dyn_cast functionality for the derived pass class. | |
| Protected Member Functions inherited from mlir::InterfacePass< mlir::FunctionOpInterface > | |
| OperationPass (TypeID passID) | |
| OperationPass (const OperationPass &)=default | |
| OperationPass (OperationPass &&)=delete | |
| bool | canScheduleOn (RegisteredOperationName opName) const final |
| Indicate if the current pass can be scheduled on the given operation type. | |
| mlir::FunctionOpInterface | getOperation () |
| Return the current operation being transformed. | |
| AnalysisT & | getAnalysis () |
| Query an analysis for the current operation. | |
| Protected Member Functions inherited from mlir::OperationPass< OpT > | |
| OperationPass (TypeID passID) | |
| OperationPass (const OperationPass &)=default | |
| OperationPass & | operator= (const OperationPass &)=delete |
| OperationPass (OperationPass &&)=delete | |
| OperationPass & | operator= (OperationPass &&)=delete |
| bool | canScheduleOn (RegisteredOperationName opName) const final |
| Indicate if the current pass can be scheduled on the given operation type. | |
| OpT | getOperation () |
| Return the current operation being transformed. | |
| template<typename AnalysisT> | |
| AnalysisT & | getAnalysis () |
| Query an analysis for the current operation of the specific derived operation type. | |
| Protected Member Functions inherited from mlir::Pass | |
| Pass (TypeID passID, std::optional< StringRef > opName=std::nullopt) | |
| Pass (const Pass &other) | |
| Pass & | operator= (const Pass &)=delete |
| Pass (Pass &&)=delete | |
| Pass & | operator= (Pass &&)=delete |
| detail::PassExecutionState & | getPassState () |
| Returns the current pass state. | |
| MLIRContext & | getContext () |
| Return the MLIR context for the current operation being transformed. | |
| virtual LogicalResult | initialize (MLIRContext *context) |
| Initialize any complex state necessary for running this pass. | |
| LogicalResult | runPipeline (OpPassManager &pipeline, Operation *op) |
| Schedule an arbitrary pass pipeline on the provided operation. | |
| std::unique_ptr< Pass > | clone () const |
| A clone method to create a copy of this pass. | |
| Operation * | getOperation () |
| Return the current operation being transformed. | |
| void | signalPassFailure () |
| Signal that some invariant was broken when running. | |
| template<typename AnalysisT> | |
| AnalysisT & | getAnalysis () |
| Query an analysis for the current ir unit. | |
| template<typename AnalysisT, typename OpT> | |
| AnalysisT & | getAnalysis () |
| Query an analysis for the current ir unit of a specific derived operation type. | |
| template<typename AnalysisT> | |
| std::optional< std::reference_wrapper< AnalysisT > > | getCachedAnalysis () |
| Query a cached instance of an analysis for the current ir unit if one exists. | |
| void | markAllAnalysesPreserved () |
| Mark all analyses as preserved. | |
| template<typename... AnalysesT> | |
| void | markAnalysesPreserved () |
| Mark the provided analyses as preserved. | |
| void | markAnalysesPreserved (TypeID id) |
| template<typename AnalysisT> | |
| std::optional< std::reference_wrapper< AnalysisT > > | getCachedParentAnalysis (Operation *parent) |
| Returns the analysis for the given parent operation if it exists. | |
| template<typename AnalysisT> | |
| std::optional< std::reference_wrapper< AnalysisT > > | getCachedParentAnalysis () |
| Returns the analysis for the parent operation if it exists. | |
| template<typename AnalysisT> | |
| std::optional< std::reference_wrapper< AnalysisT > > | getCachedChildAnalysis (Operation *child) |
| Returns the analysis for the given child operation if it exists. | |
| template<typename AnalysisT> | |
| AnalysisT & | getChildAnalysis (Operation *child) |
| Returns the analysis for the given child operation, or creates it if it doesn't exist. | |
| template<typename AnalysisT, typename OpTy> | |
| AnalysisT & | getChildAnalysis (OpTy child) |
| Returns the analysis for the given child operation of specific derived operation type, or creates it if it doesn't exist. | |
| AnalysisManager | getAnalysisManager () |
| Returns the current analysis manager. | |
| void | copyOptionValuesFrom (const Pass *other) |
| Copy the option values from 'other', which is another instance of this pass. | |
| Static Protected Member Functions inherited from mlir::OperationPass< OpT > | |
| static bool | classof (const Pass *pass) |
| Support isa/dyn_cast functionality. | |
| Protected Attributes inherited from mlir::shard::impl::ShardingPropagationBase< ShardingPropagation > | |
| ::mlir::Pass::Option< mlir::shard::TraversalOrder > | traversal |
Detailed Description
Definition at line 358 of file ShardingPropagation.cpp.
Member Function Documentation
◆ runOnOperation()
|
inlineoverridevirtual |
The polymorphic API that runs the pass over the currently held operation.
Implements mlir::Pass.
Definition at line 363 of file ShardingPropagation.cpp.
References mlir::shard::Backward, mlir::shard::BackwardForward, DBGS, mlir::shard::ForwardBackward, mlir::Region::front(), mlir::InterfacePass< mlir::FunctionOpInterface >::getOperation(), mlir::Block::getOperations(), mlir::Region::hasOneBlock(), mlir::Pass::signalPassFailure(), mlir::shard::impl::ShardingPropagationBase< ShardingPropagation >::traversal, mlir::verify(), and visitOp().
The documentation for this struct was generated from the following file:
- lib/Dialect/Shard/Transforms/ShardingPropagation.cpp
Generated on for MLIR by
1.14.0