This is an archive of the discontinued LLVM Phabricator instance.

Show First 20 Lines • Show All 70 Lines • ▼ Show 20 Lines	struct DimOfShapedTypeOpInterface : public OpRewritePattern<OpTy> {
}		}
};		};

/// Fold dim of an operation that implements the InferShapedTypeOpInterface		/// Fold dim of an operation that implements the InferShapedTypeOpInterface
template <typename OpTy>		template <typename OpTy>
struct DimOfReifyRankedShapedTypeOpInterface : public OpRewritePattern<OpTy> {		struct DimOfReifyRankedShapedTypeOpInterface : public OpRewritePattern<OpTy> {
using OpRewritePattern<OpTy>::OpRewritePattern;		using OpRewritePattern<OpTy>::OpRewritePattern;

		void initialize() { OpRewritePattern<OpTy>::setHasBoundedRewriteRecursion(); }

LogicalResult matchAndRewrite(OpTy dimOp,		LogicalResult matchAndRewrite(OpTy dimOp,
PatternRewriter &rewriter) const override {		PatternRewriter &rewriter) const override {
OpResult dimValue = dimOp.getSource().template dyn_cast<OpResult>();		OpResult dimValue = dimOp.getSource().template dyn_cast<OpResult>();
if (!dimValue)		if (!dimValue)
return failure();		return failure();
auto rankedShapeTypeOp =		auto rankedShapeTypeOp =
dyn_cast<ReifyRankedShapedTypeOpInterface>(dimValue.getOwner());		dyn_cast<ReifyRankedShapedTypeOpInterface>(dimValue.getOwner());
if (!rankedShapeTypeOp)		if (!rankedShapeTypeOp)
▲ Show 20 Lines • Show All 84 Lines • Show Last 20 Lines

[mlir] Set pattern that resolves tensor dimensions as having bounded recursion.ClosedPublic