This is an archive of the discontinued LLVM Phabricator instance.

	Show All 23 Lines
	//===----------------------------------------------------------------------===//			//===----------------------------------------------------------------------===//
	// These patterns are exposed individually because they are expected to be			// These patterns are exposed individually because they are expected to be
	// typically used individually.			// typically used individually.

	// Create a new call to the type-canonicalized `LinalgOp::getLibraryCallName()`			// Create a new call to the type-canonicalized `LinalgOp::getLibraryCallName()`
	// function. The implementation of the function can be either in the same module			// function. The implementation of the function can be either in the same module
	// or in an externally linked library.			// or in an externally linked library.
	// This is a generic entry point for all LinalgOp, except for CopyOp and			// This is a generic entry point for all LinalgOp, except for CopyOp and
	// IndexedGenericOp, for which omre specialized patterns are provided.			// IndexedGenericOp, for which more specialized patterns are provided.
	class LinalgOpToLibraryCallRewrite			class LinalgOpToLibraryCallRewrite
	: public OpInterfaceRewritePattern<LinalgOp> {			: public OpInterfaceRewritePattern<LinalgOp> {
	public:			public:
	using OpInterfaceRewritePattern<LinalgOp>::OpInterfaceRewritePattern;			using OpInterfaceRewritePattern<LinalgOp>::OpInterfaceRewritePattern;

	LogicalResult matchAndRewrite(LinalgOp op,			LogicalResult matchAndRewrite(LinalgOp op,
	PatternRewriter &rewriter) const override;			PatternRewriter &rewriter) const override;
	};			};
	▲ Show 20 Lines • Show All 41 Lines • Show Last 20 Lines

mlir/lib/Conversion/LinalgToStandard/LinalgToStandard.cpp

	Show First 20 Lines • Show All 100 Lines • ▼ Show 20 Lines
	}			}

	LogicalResult mlir::linalg::LinalgOpToLibraryCallRewrite::matchAndRewrite(			LogicalResult mlir::linalg::LinalgOpToLibraryCallRewrite::matchAndRewrite(
	LinalgOp op, PatternRewriter &rewriter) const {			LinalgOp op, PatternRewriter &rewriter) const {
	// Only LinalgOp for which there is no specialized pattern go through this.			// Only LinalgOp for which there is no specialized pattern go through this.
	if (isa<CopyOp>(op) \|\| isa<IndexedGenericOp>(op))			if (isa<CopyOp>(op) \|\| isa<IndexedGenericOp>(op))
	return failure();			return failure();

	// TODO: remove once index ops are supported.
	if (op.hasIndexSemantics())
	return failure();

	auto libraryCallName = getLibraryCallSymbolRef(op, rewriter);			auto libraryCallName = getLibraryCallSymbolRef(op, rewriter);
	if (!libraryCallName)			if (!libraryCallName)
	return failure();			return failure();

				// TODO: Add support for more complex library call signatures that include
				// indices or captured values.
	rewriter.replaceOpWithNewOp<mlir::CallOp>(			rewriter.replaceOpWithNewOp<mlir::CallOp>(
	op, libraryCallName.getValue(), TypeRange(),			op, libraryCallName.getValue(), TypeRange(),
	createTypeCanonicalizedMemRefOperands(rewriter, op->getLoc(),			createTypeCanonicalizedMemRefOperands(rewriter, op->getLoc(),
	op->getOperands()));			op->getOperands()));
	return success();			return success();
	}			}

	LogicalResult mlir::linalg::CopyOpToLibraryCallRewrite::matchAndRewrite(			LogicalResult mlir::linalg::CopyOpToLibraryCallRewrite::matchAndRewrite(
	▲ Show 20 Lines • Show All 110 Lines • Show Last 20 Lines

This is an archive of the discontinued LLVM Phabricator instance.

[mlir][linalg] enable library call rewrites for linalg operations with index semantics.ClosedPublic

Details

Diff Detail

Event Timeline

Revision Contents

Diff 338498

mlir/include/mlir/Conversion/LinalgToStandard/LinalgToStandard.h

mlir/lib/Conversion/LinalgToStandard/LinalgToStandard.cpp

[mlir][linalg] enable library call rewrites for linalg operations with index semantics.
ClosedPublic