[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