Open
Description
There are a few places in cuML where we are using UCX endpoints directly to perform scatter and gather calls. We originally kept the API minimal, since users could compose their own scatter and gather calls out of the isend
and irecv
calls.
However, this can be error prone and we're already finding small subtle places where we diverged from the MPI spec and this caused unexpected behaviors. While I don't think we should claim to adhere strictly to the MPI spec, I do think that including the scatter and gather calls in the comms would alleviate some of this, since they are common communications primitives.
Metadata
Assignees
Type
Projects
Status
No status