Skip to content

Forward-merge release/0.50 into main#654

Merged
AyodeAwe merged 1 commit into
mainfrom
release/0.50
May 19, 2026
Merged

Forward-merge release/0.50 into main#654
AyodeAwe merged 1 commit into
mainfrom
release/0.50

Conversation

@rapids-bot
Copy link
Copy Markdown

@rapids-bot rapids-bot Bot commented May 19, 2026

Forward-merge triggered by push to release/0.50 that creates a PR to keep main up-to-date. If this PR is unable to be immediately merged due to conflicts, it will remain open for the team to manually merge. See forward-merger docs for more info.

Add an opt-in mechanism for querying UCP request attributes (memory type, debug string) on every `ucxx::Request`. Enabled via a new `ucxx::experimental::WorkerBuilder::requestAttributes(true)` option. When enabled, every `ucxx::Request` submit site funnels through a small `Request::publishRequest()` helper that stores the UCP handle and queries `ucp_request_query` under the existing `_mutex`. `ucp_request_free` moves from `Request::callback` into `Request::setStatus`, making the query and the free mutually exclusive without any new atomics or callback-side locking. Wired through every request type and exposed to users via `Request::queryAttributes()`, which throws `ucxx::UnsupportedError` when the feature is disabled on the owning worker and `ucxx::NoElemError` when UCX took an inline path that produced no UCP handle to query (e.g., an eager UCX transfer).

Tag, AM, and MemoryGet test are asserted strictly above the rendezvous threshold, where UCX deterministically allocates a queryable request on every transport. Stream and MemoryPut use lenient assertions (substring-check on success, throw is acceptable) because stream has no rendezvous protocol and small RMA puts are fire-and-forget, both of which have transport-dependent inline-completion behavior that no fixed size threshold can portably predict.

The toggle is worker-scoped: enabling it queries attributes on **every** request created from that worker, which has potentially non-negligible per-request cost. Fine-grained per-request opt-in (so callers can attribute-query only the requests they care about) is not implemented here, it requires a builder-pattern constructor at the request level which doesn't exist yet, and is deferred to a follow-up. For now, users who need attributes accept the worker-wide cost, and users who don't, opt out by leaving the default.

Authors:
  - Peter Andreas Entschev (https://github.com/pentschev)

Approvers:
  - Lawrence Mitchell (https://github.com/wence-)
  - Bradley Dice (https://github.com/bdice)

URL: #437
@rapids-bot rapids-bot Bot requested review from a team as code owners May 19, 2026 12:53
@rapids-bot rapids-bot Bot requested a review from bdice May 19, 2026 12:53
@AyodeAwe AyodeAwe merged commit e2564ad into main May 19, 2026
@rapids-bot
Copy link
Copy Markdown
Author

rapids-bot Bot commented May 19, 2026

SUCCESS - forward-merge complete.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants