Skip to content

Add support for AM in Python benchmark #393

@pentschev

Description

@pentschev

Currently the Python send_recv benchmark does not support AM, this needs to be added. Furthermore, in UCX-Py there is an inconsistency with receiving AM when compared to ucx_perftest, where UCX-Py always reallocates memory regardless of the --reuse-alloc being specified, see rapidsai/ucx-py#1125 (comment) for more details, a way to handle that is to register an AM allocator that always returns the same memory buffer.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions