[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