[Mlir-commits] [mlir] [mlir][mesh] Add resharding spmdization on a 1D device mesh (PR #76179)

Chengji Yao llvmlistbot at llvm.org
Mon Jan 1 12:54:15 PST 2024


================
@@ -186,6 +209,29 @@ def Mesh_ShardOp : Mesh_Op<"shard", [Pure, SameOperandsAndResultType]> {
   }];
 }
 
+def Mesh_ProcessIndexOp : Mesh_Op<"process_index", [Pure, DeclareOpInterfaceMethods<SymbolUserOpInterface>]> {
----------------
yaochengji wrote:

Is there an Int value representing the device/process id, not a list of int?

https://github.com/llvm/llvm-project/pull/76179


More information about the Mlir-commits mailing list