XeGPUContiguityAnalysis.cpp File Reference

#include "mlir/Analysis/DataFlow/DeadCodeAnalysis.h"
#include "mlir/Analysis/DataFlow/SparseAnalysis.h"
#include "mlir/Analysis/DataFlowFramework.h"
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/MemRef/IR/MemRef.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "mlir/Dialect/XeGPU/IR/XeGPU.h"
#include "mlir/Dialect/XeGPU/Transforms/Passes.h"
#include "mlir/Dialect/XeGPU/Transforms/Transforms.h"
#include "mlir/Dialect/XeGPU/Utils/XeGPUUtils.h"
#include "mlir/IR/BuiltinAttributes.h"
#include "mlir/IR/BuiltinTypes.h"
#include "llvm/ADT/APInt.h"
#include "llvm/Support/MathExtras.h"
#include <numeric>
#include <optional>

Go to the source code of this file.

Classes
struct	mlir::xegpu::detail::axis_dataflow::AxisInfo
	Per-dimension axis information for an SSA vector value of integer / index type. More...
class	mlir::xegpu::detail::axis_dataflow::AxisInfoAnalysis
	Sparse forward dataflow analysis that computes AxisInfo for vector values reachable from the entry of the analyzed op. More...

Namespaces
namespace	mlir
	Include the generated interface declarations.
namespace	mlir::xegpu
namespace	mlir::xegpu::detail
namespace	mlir::xegpu::detail::axis_dataflow

Macros
#define	DEBUG_TYPE   "xegpu-contiguity-analysis"

Typedefs
using	mlir::xegpu::detail::axis_dataflow::AxisInfoLattice = dataflow::Lattice<AxisInfo>

Functions
static int64_t	mlir::xegpu::detail::axis_dataflow::highestPow2Divisor (int64_t v)
	Power-of-two divisor of v. Returns kAxisInfoTop when v == 0.
static AxisInfo	mlir::xegpu::detail::axis_dataflow::entryStateFor (Value v)
	Initial lattice value for an SSA value when no transfer function applies.
static AxisInfo	mlir::xegpu::detail::axis_dataflow::splatAxisInfo (ArrayRef< int64_t > shape, int64_t c)
	AxisInfo for a tensor that is constant c everywhere.

Variables
static constexpr int64_t	mlir::xegpu::detail::axis_dataflow::kAxisInfoTop = 1LL << 30
	Sentinel "very large" value for unconstrained dimensions.

Macro Definition Documentation

DEBUG_TYPE

#define DEBUG_TYPE   "xegpu-contiguity-analysis"

Definition at line 41 of file XeGPUContiguityAnalysis.cpp.