This is an archive of the discontinued LLVM Phabricator instance.

	Show First 20 Lines • Show All 169 Lines • ▼ Show 20 Lines

	LogicalResult			LogicalResult
	ShapeOfOpConverter::matchAndRewrite(ShapeOfOp op, ArrayRef<Value> operands,			ShapeOfOpConverter::matchAndRewrite(ShapeOfOp op, ArrayRef<Value> operands,
	ConversionPatternRewriter &rewriter) const {			ConversionPatternRewriter &rewriter) const {
	ShapeOfOp::Adaptor transformed(operands);			ShapeOfOp::Adaptor transformed(operands);
	Value arg = transformed.arg();			Value arg = transformed.arg();
	Type argTy = arg.getType();			Type argTy = arg.getType();

				// For now, this lowering supports only error-free arguments.
				if (argTy.isa<ValueShapeType>())
				return failure();

	// For ranked tensors `shape_of` lowers to `std` and the pattern can be			// For ranked tensors `shape_of` lowers to `std` and the pattern can be
	// found in the corresponding pass.			// found in the corresponding pass.
	if (argTy.isa<RankedTensorType>())			if (argTy.isa<RankedTensorType>())
	return failure();			return failure();

	// Allocate stack memory.			// Allocate stack memory.
	auto loc = op.getLoc();			auto loc = op.getLoc();
	Value rank = rewriter.create<mlir::RankOp>(loc, arg);			Value rank = rewriter.create<mlir::RankOp>(loc, arg);
	▲ Show 20 Lines • Show All 60 Lines • Show Last 20 Lines

[MLIR][Shape] Limit shape to SCF lowering patterns to their supported typesClosedPublic