This is an archive of the discontinued LLVM Phabricator instance.

	Show First 20 Lines • Show All 395 Lines • ▼ Show 20 Lines
	mlir::computeRankReductionMask(ArrayRef<int64_t> originalShape,			mlir::computeRankReductionMask(ArrayRef<int64_t> originalShape,
	ArrayRef<int64_t> reducedShape) {			ArrayRef<int64_t> reducedShape) {
	size_t originalRank = originalShape.size(), reducedRank = reducedShape.size();			size_t originalRank = originalShape.size(), reducedRank = reducedShape.size();
	llvm::SmallDenseSet<unsigned> unusedDims;			llvm::SmallDenseSet<unsigned> unusedDims;
	unsigned reducedIdx = 0;			unsigned reducedIdx = 0;
	for (unsigned originalIdx = 0; originalIdx < originalRank; ++originalIdx) {			for (unsigned originalIdx = 0; originalIdx < originalRank; ++originalIdx) {
	// Greedily insert `originalIdx` if match.			// Greedily insert `originalIdx` if match.
	if (reducedIdx < reducedRank &&			if (reducedIdx < reducedRank &&
	originalShape[originalIdx] == reducedShape[reducedIdx]) {			(originalShape[originalIdx] == reducedShape[reducedIdx] \|\|
				originalShape[originalIdx] == ShapedType::kDynamic \|\|
				reducedShape[reducedIdx] == ShapedType::kDynamic)) {
	reducedIdx++;			reducedIdx++;
	continue;			continue;
	}			}

	unusedDims.insert(originalIdx);			unusedDims.insert(originalIdx);
	// If no match on `originalIdx`, the `originalShape` at this dimension			// If no match on `originalIdx`, the `originalShape` at this dimension
	// must be 1, otherwise we bail.			// must be 1, otherwise we bail.
	if (originalShape[originalIdx] != 1)			if (originalShape[originalIdx] != 1)
	▲ Show 20 Lines • Show All 527 Lines • Show Last 20 Lines

This is an archive of the discontinued LLVM Phabricator instance.

[mlir] Relaxing validity of dimensions of slicesAbandonedPublic

Details

Diff Detail

Event Timeline

Revision Contents

Diff 503609

mlir/lib/IR/BuiltinTypes.cpp

[mlir] Relaxing validity of dimensions of slices
AbandonedPublic