[Mlir-commits] [mlir] [mlir][mesh] Use tensor shape notation for the shape of a cluster (PR #73826)
Mehdi Amini
llvmlistbot at llvm.org
Fri Dec 1 22:31:09 PST 2023
================
@@ -40,26 +40,27 @@ def Mesh_ClusterOp : Mesh_Op<"cluster", [Symbol]> {
determine the layout and the addressing space of the computation distributed
across the mesh.
- 3. `dim_sizes`: This attribute represents the device assignment along the
- axes of the cluster. Each integer in the array corresponds to the number of
- devices along a specific axis. If an integer value is 0, it implies that the
- number of devices along that axis is unknown. This flexibility allows for
- dynamic device assignment or configurations where the exact number of
- devices might not be determined during compile time.
+ 3. `dim_sizes`: This attribute represents the shape of the device cluster.
+ It uses the same notation as a tensor shape. Also allowing for dynamic
----------------
joker-eph wrote:
If it's not about the formatting, I rather keep changes separate if possible.
https://github.com/llvm/llvm-project/pull/73826
More information about the Mlir-commits
mailing list