================ @@ -529,6 +519,124 @@ struct ConvertShardShapeOp : public OpConversionPattern<ShardShapeOp> { } }; +static mpi::MPI_OpClassEnumAttr getMPIReduction(ReductionKindAttr kind) { ---------------- Dinistro wrote: Nit: Use a more descriptive name. https://github.com/llvm/llvm-project/pull/144060