@@ -1504,8 +1504,9 @@ struct VectorExtractStridedSliceDistribution
15041504 auto distributedType =
15051505 cast<VectorType>(warpOp.getResult (operandIdx).getType ());
15061506 // Find the distributed dimension. There should be exactly one.
1507- auto yieldedType = cast<VectorType>(operand->get ().getType ());
1508- auto distributedDims = getDistributedDims (yieldedType, distributedType);
1507+ auto extractResultType = cast<VectorType>(operand->get ().getType ());
1508+ auto distributedDims =
1509+ getDistributedDims (extractResultType, distributedType);
15091510 // Only single dimension distribution is supported.
15101511 if (distributedDims.size () != 1 )
15111512 return rewriter.notifyMatchFailure (
@@ -1616,8 +1617,9 @@ struct VectorInsertStridedSliceDistribution
16161617 cast<VectorType>(warpOp.getResult (operandNumber).getType ());
16171618 // Find the distributed dimension of the dest vector. There should be
16181619 // exactly one.
1619- auto yieldedType = cast<VectorType>(operand->get ().getType ());
1620- auto destDistributedDims = getDistributedDims (yieldedType, distributedType);
1620+ auto insertResultType = cast<VectorType>(operand->get ().getType ());
1621+ auto destDistributedDims =
1622+ getDistributedDims (insertResultType, distributedType);
16211623 // Only single dimension distribution is supported.
16221624 if (destDistributedDims.size () != 1 )
16231625 return rewriter.notifyMatchFailure (
0 commit comments