================ @@ -529,6 +519,124 @@ struct ConvertShardShapeOp : public OpConversionPattern<ShardShapeOp> { } }; +static mpi::MPI_OpClassEnumAttr getMPIReduction(ReductionKindAttr kind) { ---------------- fschlimb wrote: like what? https://github.com/llvm/llvm-project/pull/144060