[Mlir-commits] [mlir] [mlir][mesh] Use tensor shape notation for the shape of a cluster (PR #73826)
Boian Petkantchin
llvmlistbot at llvm.org
Fri Dec 1 16:27:50 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
----------------
sogartar wrote:
I think we can remove the `rank` attribute. I don't envision a scenario where we would have to describe a cluster shape with many trailing dynamic dimensions like
```
2x3x?x?x?x?x?x?x?x?
```
Famous last words.
Should I remove it in this PR?
https://github.com/llvm/llvm-project/pull/73826
More information about the Mlir-commits
mailing list