MLIR 23.0.0git
mlir::xegpu::detail::axis_dataflow::AxisInfoAnalysis Class Reference

Sparse forward dataflow analysis that computes AxisInfo for vector values reachable from the entry of the analyzed op. More...

Inheritance diagram for mlir::xegpu::detail::axis_dataflow::AxisInfoAnalysis:

Public Member Functions

LogicalResult visitOperation (Operation *op, ArrayRef< const AxisInfoLattice * > operands, ArrayRef< AxisInfoLattice * > results) override
void setToEntryState (AxisInfoLattice *lattice) override
Public Member Functions inherited from mlir::dataflow::SparseForwardDataFlowAnalysis< AxisInfoLattice >
 SparseForwardDataFlowAnalysis (DataFlowSolver &solver)
virtual LogicalResult visitOperation (Operation *op, ArrayRef< const AxisInfoLattice * > operands, ArrayRef< AxisInfoLattice * > results)=0
 Visit an operation with the lattices of its operands.
virtual void visitExternalCall (CallOpInterface call, ArrayRef< const AxisInfoLattice * > argumentLattices, ArrayRef< AxisInfoLattice * > resultLattices)
 Visit a call operation to an externally defined function given the lattices of its arguments.
virtual void visitNonControlFlowArguments (Operation *op, const RegionSuccessor &successor, ValueRange nonSuccessorInputs, ArrayRef< AxisInfoLattice * > nonSuccessorInputLattices)
 Given an operation with possible region control-flow, the lattices of the operands, and a region successor, compute the lattice values for non-successor-inputs (ex.
Public Member Functions inherited from mlir::dataflow::AbstractSparseForwardDataFlowAnalysis
LogicalResult initialize (Operation *top) override
 Initialize the analysis by visiting every owner of an SSA value: all operations and blocks.
LogicalResult visit (ProgramPoint *point) override
 Visit a program point.
Public Member Functions inherited from mlir::DataFlowAnalysis
virtual ~DataFlowAnalysis ()
 DataFlowAnalysis (DataFlowSolver &solver)
 Create an analysis with a reference to the parent solver.
virtual void initializeEquivalentLatticeAnchor (Operation *top)
 Initialize lattice anchor equivalence class from the provided top-level operation.
TypeID getTypeID () const
 Return the TypeID of the concrete analysis class.

Additional Inherited Members

Protected Member Functions inherited from mlir::dataflow::SparseForwardDataFlowAnalysis< AxisInfoLattice >
AxisInfoLattice * getLatticeElement (Value value) override
 Get the lattice element for a value.
const AxisInfoLattice * getLatticeElementFor (ProgramPoint *point, Value value)
 Get the lattice element for a value and create a dependency on the provided program point.
virtual void setToEntryState (AxisInfoLattice *lattice)=0
 Set the given lattice element(s) at control flow entry point(s).
void setAllToEntryStates (ArrayRef< AxisInfoLattice * > lattices)
Protected Member Functions inherited from mlir::dataflow::AbstractSparseForwardDataFlowAnalysis
 AbstractSparseForwardDataFlowAnalysis (DataFlowSolver &solver)
const AbstractSparseLatticegetLatticeElementFor (ProgramPoint *point, Value value)
 Get a read-only lattice element for a value and add it as a dependency to a program point.
virtual void setToEntryState (AbstractSparseLattice *lattice)=0
 Set the given lattice element(s) at control flow entry point(s).
void setAllToEntryStates (ArrayRef< AbstractSparseLattice * > lattices)
void join (AbstractSparseLattice *lhs, const AbstractSparseLattice &rhs)
 Join the lattice element and propagate and update if it changed.
virtual LogicalResult visitCallOperation (CallOpInterface call, ArrayRef< const AbstractSparseLattice * > operandLattices, ArrayRef< AbstractSparseLattice * > resultLattices)
 Visits a call operation.
virtual void visitCallableOperation (CallableOpInterface callable, ArrayRef< AbstractSparseLattice * > argLattices)
 Visits a callable operation.
Protected Member Functions inherited from mlir::DataFlowAnalysis
void addDependency (AnalysisState *state, ProgramPoint *point)
 Create a dependency between the given analysis state and lattice anchor on this analysis.
void propagateIfChanged (AnalysisState *state, ChangeResult changed)
 Propagate an update to a state if it changed.
template<typename AnchorT>
void registerAnchorKind ()
 Register a custom lattice anchor class.
template<typename AnchorT, typename... Args>
AnchorT * getLatticeAnchor (Args &&...args)
 Get or create a custom lattice anchor.
template<typename StateT, typename AnchorT>
void unionLatticeAnchors (AnchorT anchor, AnchorT other)
 Union input anchors under the given state.
template<typename StateT, typename AnchorT>
StateT * getOrCreate (AnchorT anchor)
 Get the analysis state associated with the lattice anchor.
template<typename StateT, typename AnchorT>
const StateT * getOrCreateFor (ProgramPoint *dependent, AnchorT anchor)
 Get a read-only analysis state for the given point and create a dependency on dependent.
ProgramPointgetProgramPointBefore (Operation *op)
 Get a uniqued program point instance.
ProgramPointgetProgramPointBefore (Block *block)
ProgramPointgetProgramPointAfter (Operation *op)
ProgramPointgetProgramPointAfter (Block *block)
const DataFlowConfiggetSolverConfig () const
 Return the configuration of the solver used for this analysis.

Detailed Description

Sparse forward dataflow analysis that computes AxisInfo for vector values reachable from the entry of the analyzed op.

Definition at line 191 of file XeGPUContiguityAnalysis.cpp.

Member Function Documentation

◆ setToEntryState()

void mlir::xegpu::detail::axis_dataflow::AxisInfoAnalysis::setToEntryState ( AxisInfoLattice * lattice)
inlineoverride

◆ visitOperation()

LogicalResult mlir::xegpu::detail::axis_dataflow::AxisInfoAnalysis::visitOperation ( Operation * op,
ArrayRef< const AxisInfoLattice * > operands,
ArrayRef< AxisInfoLattice * > results )
inlineoverride

Definition at line 197 of file XeGPUContiguityAnalysis.cpp.

References add, div, mul, rem, and success().


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