This is an archive of the discontinued LLVM Phabricator instance.

Show First 20 Lines • Show All 361 Lines • ▼ Show 20 Lines	LogicalResult matchAndRewrite(tensor::CastOp op,
if (srcType == dstType) {		if (srcType == dstType) {
rewriter.replaceOp(op, op->getResults());		rewriter.replaceOp(op, op->getResults());
return success();		return success();
}		}
// See if a sparsity changing cast can be fused into producer.		// See if a sparsity changing cast can be fused into producer.
if (tensor::isSameTypeWithoutEncoding(srcType, dstType)) {		if (tensor::isSameTypeWithoutEncoding(srcType, dstType)) {
if (Operation *def = op.getSource().getDefiningOp()) {		if (Operation *def = op.getSource().getDefiningOp()) {
if (def->hasOneUse() && isa<tensor::ExtractSliceOp>(def)) {		if (def->hasOneUse() && isa<tensor::ExtractSliceOp>(def)) {
		rewriter.updateRootInPlace(def, [&]() {
def->getResult(0).setType(op->getResultTypes()[0]);		def->getResult(0).setType(op->getResultTypes()[0]);
		});
rewriter.replaceOp(op, def->getResult(0));		rewriter.replaceOp(op, def->getResult(0));
return success();		return success();
}		}
}		}
}		}
// Repair tensor casts with at least one sparse operand into the		// Repair tensor casts with at least one sparse operand into the
// the properly supported sparse_tensor.convert.		// the properly supported sparse_tensor.convert.
if (getSparseTensorEncoding(srcType) \|\| getSparseTensorEncoding(dstType)) {		if (getSparseTensorEncoding(srcType) \|\| getSparseTensorEncoding(dstType)) {
▲ Show 20 Lines • Show All 906 Lines • Show Last 20 Lines

[mlir][sparse] Fix incorrect API usage in RewritePatternsClosedPublic