Right now, communication buffers in UCXBufferCommunicator is registered by the first send/recv operation during warmup. A better strategy would register these buffers directly through UCXCommunicator::register_buffer after https://github.com/rapidsai/distributed-join/pull/36 is merged.
Right now, communication buffers in
UCXBufferCommunicatoris registered by the first send/recv operation during warmup. A better strategy would register these buffers directly throughUCXCommunicator::register_bufferafter https://github.com/rapidsai/distributed-join/pull/36 is merged.