Skip to content

Commit 834094f

Browse files
committed
Update warp_execute_on_lane_0 description
1 parent 7d5b1b4 commit 834094f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

mlir/include/mlir/Dialect/Vector/IR/VectorOps.td

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3000,7 +3000,8 @@ def Vector_WarpExecuteOnLane0Op : Vector_Op<"warp_execute_on_lane_0",
30003000

30013001
Return values are distributed on all lanes using laneId as index. The
30023002
vector is distributed based on the shape ratio between the vector type of
3003-
the yield and the result type.
3003+
the yield and the result type. Any ShapedType return value is allowed to be
3004+
distributed to support other vector-like types (e.g., xegpu.tensor_desc).
30043005
If the shapes are the same this means the value is broadcasted to all lanes.
30053006
In the future the distribution can be made more explicit using affine_maps
30063007
and will support having multiple Ids.

0 commit comments

Comments
 (0)