This is an archive of the discontinued LLVM Phabricator instance.

Show First 20 Lines • Show All 238 Lines • ▼ Show 20 Lines	matchAndRewrite(tensor::DimOp op, ArrayRef<Value> operands,
auto enc = getSparseTensorEncoding(op.source().getType());		auto enc = getSparseTensorEncoding(op.source().getType());
if (!enc)		if (!enc)
return failure();		return failure();
// Permute the dim index.		// Permute the dim index.
Optional<int64_t> index = op.getConstantIndex();		Optional<int64_t> index = op.getConstantIndex();
if (!index.hasValue())		if (!index.hasValue())
return failure();		return failure();
int64_t idx = index.getValue();		int64_t idx = index.getValue();
AffineMap p = enc.getDimOrdering();		if (AffineMap p = enc.getDimOrdering())
if (p) {		idx = p.getPermutedPosition(idx);
assert(p.isPermutation());
for (unsigned i = 0, sz = p.getNumResults(); i < sz; i++) {
if (p.getDimPosition(i) == idx) {
idx = i;
break;
}
}
}
// Generate the call.		// Generate the call.
StringRef name = "sparseDimSize";		StringRef name = "sparseDimSize";
SmallVector<Value, 2> params;		SmallVector<Value, 2> params;
params.push_back(operands[0]);		params.push_back(operands[0]);
params.push_back(		params.push_back(
rewriter.create<ConstantOp>(op.getLoc(), rewriter.getIndexAttr(idx)));		rewriter.create<ConstantOp>(op.getLoc(), rewriter.getIndexAttr(idx)));
rewriter.replaceOpWithNewOp<CallOp>(		rewriter.replaceOpWithNewOp<CallOp>(
op, resType, getFunc(op, name, resType, params), params);		op, resType, getFunc(op, name, resType, params), params);
▲ Show 20 Lines • Show All 210 Lines • Show Last 20 Lines

[mlir][sparse] use new permutation utility to avoid codedupClosedPublic