Skip to content

UCM: Propagate memory attributes in allocation events - v1.22.x - #11655

Merged
gleon99 merged 2 commits into
openucx:v1.22.xfrom
gleon99:agent/ucm-mem-event-attributes-1.22
Jul 15, 2026
Merged

UCM: Propagate memory attributes in allocation events - v1.22.x#11655
gleon99 merged 2 commits into
openucx:v1.22.xfrom
gleon99:agent/ucm-mem-event-attributes-1.22

Conversation

@gleon99

@gleon99 gleon99 commented Jul 15, 2026

Copy link
Copy Markdown
Contributor

What?

Port #11646

Comment thread src/uct/cuda/cuda_copy/cuda_copy_ep.c Outdated
if (ucs_unlikely((status == UCS_ERR_UNSUPPORTED) ||
(mem_info.type == UCS_MEMORY_TYPE_UNKNOWN))) {
(mem_info.type == UCS_MEMORY_TYPE_UNKNOWN) ||
(mem_info.sys_dev == UCS_SYS_DEVICE_ID_UNKNOWN))) {

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Just noticed this, shouldn't it have && mem_info.mem_flags == 0 here?

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Intentional — CUDA copy needs sys_dev regardless of mem_flags. We actually removed that check in the original PR after review, so I’d keep the backport consistent with master.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

CUDA copy needs sys_dev regardless of mem_flags

Why? Can it not lead to excessive queries?

We actually removed that check in the original PR after review, so I’d keep the backport consistent with master.

Sure, but just want to make sure this is not an issue later

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

👍🏼

@shasson5

Copy link
Copy Markdown
Contributor

@gleon99 can we merge?

@gleon99

gleon99 commented Jul 15, 2026

Copy link
Copy Markdown
Contributor Author

@gleon99 can we merge?

Pushed last minor change @tomerg-nvidia

@gleon99
gleon99 merged commit 5c3ef8d into openucx:v1.22.x Jul 15, 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