Skip to content

[WIP-DNM] UCT/IB/MLX5: Purge outstanding AM bcopy operations - #9

Draft
roiedanino wants to merge 1 commit into
uct-purge-am-short-bcopyfrom
uct-purge-am-bcopy
Draft

roiedanino wants to merge 1 commit into
uct-purge-am-short-bcopyfrom
uct-purge-am-bcopy

Conversation

@roiedanino

Copy link
Copy Markdown
Owner

What?

Add native RC mlx5 outstanding-purge support for uct_ep_am_bcopy() through uct_rc_mlx5_op_info_fill_am_bcopy(), with small- and large-payload cases using the existing UCT outstanding-purge fixture.

This is a stacked follow-up to openucx/ucx#11857. The base branch keeps the incremental diff limited to AM bcopy.

Why?

Fault-tolerance handling needs the AM ID, flags, and buffered payload for outstanding AM bcopy operations, while safely releasing their transport-owned send buffers.

How?

Recognize the single-data-segment SEND layout and match it to its outstanding bcopy descriptor. Use the existing MTU-based packet-count helper and PSN tracking to select undelivered operations. Keep the payload valid through the purge callback, then release the descriptor before completing subsequent flushes.

Reuse the shared endpoint invalidation, token query, purge, and completion flow. Extend AM field and payload comparisons for bcopy without introducing a separate transport-specific fixture.

No send-path changes, new PSN tracking, AM zcopy, RMA, DM, or flush-operation reporting are included.

@roiedanino roiedanino added the WIP-DNM Work in progress - do not merge label Sep 15, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

WIP-DNM Work in progress - do not merge

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant