Skip to content

UCM: Propagate memory attributes in allocation events - #11646

Merged
gleon99 merged 3 commits into
openucx:masterfrom
gleon99:agent/ucm-mem-event-attributes
Jul 14, 2026
Merged

gleon99 merged 3 commits into
openucx:masterfrom
gleon99:agent/ucm-mem-event-attributes

Conversation

@gleon99

@gleon99 gleon99 commented Jul 10, 2026 •

Copy link
Copy Markdown
Contributor

What?

Propagate sys_dev and mem_flags in UCM memory events and remove UCS_MEM_FLAG_NEEDS_QUERY.

Why?

Device symbols require additional memory-attribute detection. Carrying attributes in the event avoids a synthetic query flag.

Fixes internal issue.

How?

Device-symbol events use CUDA-managed type, unknown sys_dev, and zero flags; other allocations are marked registerable.

@gleon99 gleon99 added WIP-DNM Work in progress / Do not review and removed WIP-DNM Work in progress / Do not review labels Jul 10, 2026
@gleon99
gleon99 requested review from brminich and rakhmets July 12, 2026 15:17
@gleon99

gleon99 commented Jul 14, 2026

Copy link
Copy Markdown
Contributor Author

@rakhmets @brminich .

@brminich
brminich requested a review from tomerg-nvidia July 14, 2026 11:57
Comment thread src/uct/cuda/cuda_copy/cuda_copy_ep.c Resolved
Comment thread test/gtest/ucs/test_memtype_cache.cc Resolved
@gleon99

gleon99 commented Jul 14, 2026

Copy link
Copy Markdown
Contributor Author

@brminich / @tomerg-nvidia asap plz

@gleon99
gleon99 enabled auto-merge (squash) July 14, 2026 19:53
@gleon99
gleon99 merged commit 09c9676 into openucx:master Jul 14, 2026
161 checks passed
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.

3 participants