Details
-
Improvement
-
Status: Resolved
-
Major
-
Resolution: Fixed
-
None
Description
We should add support for sending/receiving data using a non-standard allocator, given that:
- the plan is to support UCX as a backend to Flight,
- UCX can manage non-CPU memory,
- the existing Device/MemoryManager API handles this case,
We should find some way to ensure we fully reflect UCX's capabilities to Flight users. Furthermore, we should integrate the MemoryManager and Flight APIs so that Flight user code should not have to worry about whether their backend supports this or not. (That means that for gRPC, we should do the copy for the user.)
As part of this, we should extend the Flight benchmark to test this case so we also have a baseline.
Attachments
Issue Links
- links to