This is an archive of the discontinued LLVM Phabricator instance.

Show First 20 Lines • Show All 1,460 Lines • ▼ Show 20 Lines	LogicalResult matchAndRewrite(WarpExecuteOnLane0Op warpOp,
SmallVector<Type> distTypes;		SmallVector<Type> distTypes;
mlir::visitUsedValuesDefinedAbove(		mlir::visitUsedValuesDefinedAbove(
forOp.getBodyRegion(), [&](OpOperand *operand) {		forOp.getBodyRegion(), [&](OpOperand *operand) {
Operation *parent = operand->get().getParentRegion()->getParentOp();		Operation *parent = operand->get().getParentRegion()->getParentOp();
if (warpOp->isAncestor(parent)) {		if (warpOp->isAncestor(parent)) {
if (!escapingValues.insert(operand->get()))		if (!escapingValues.insert(operand->get()))
return;		return;
Type distType = operand->get().getType();		Type distType = operand->get().getType();
if (auto vecType = cast<VectorType>(distType)) {		if (auto vecType = dyn_cast<VectorType>(distType)) {
AffineMap map = distributionMapFn(operand->get());		AffineMap map = distributionMapFn(operand->get());
distType = getDistributedType(vecType, map, warpOp.getWarpSize());		distType = getDistributedType(vecType, map, warpOp.getWarpSize());
}		}
inputTypes.push_back(operand->get().getType());		inputTypes.push_back(operand->get().getType());
distTypes.push_back(distType);		distTypes.push_back(distType);
}		}
});		});

▲ Show 20 Lines • Show All 228 Lines • Show Last 20 Lines

[mlir][vector] Use dyn_cast in if conditionsClosedPublic