Skip to content

Commit 5d33f84

Browse files
committed
feedback
1 parent 1e07368 commit 5d33f84

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

mlir/lib/Dialect/XeGPU/Transforms/XeGPUSubgroupDistribute.cpp

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)