Mojo function
allreduce
allreduce[dtype: DType, rank: Int, ngpus: Int](inputs: InlineArray[NDBuffer[dtype, rank, MutAnyOrigin], ngpus], outputs: InlineArray[NDBuffer[dtype, rank, MutAnyOrigin], ngpus], list_of_ctx: List[DeviceContext])
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!