UCM/ZE/TEST: Add ZE support to gtest mem_buffer and UCM hook tests#11321
Open
yafshar wants to merge 4 commits into
Open
UCM/ZE/TEST: Add ZE support to gtest mem_buffer and UCM hook tests#11321yafshar wants to merge 4 commits into
yafshar wants to merge 4 commits into
Conversation
- Extend gtest mem_buffer with ZE host, device, and managed allocators, including ZE copy and memset paths, and centralize ZE runtime initialization. - Add UCM ZE hook tests to validate alloc and free callbacks. - Update gtest Makefile.am to include ZE sources and linker flags.
Contributor
Author
|
Hi @yosefe, just pinging for visibility. It’s been open for a few weeks and I wanted to check if you had a chance to look, or if there’s anything I can do to help move it forward. |
Contributor
Author
|
Hi @yosefe, just a gentle follow‑up in case this slipped through. |
yafshar
force-pushed
the
test/ze-mem-buffer-ucm-hooks
branch
from
01288fb to
072a8d0
Compare
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
What?
Add Level Zero (ZE) memory support to gtest mem_buffer and introduce UCM ZE hook tests.
This PR:
Why?
UCX gtests already cover GPU memory handling and UCM hook behavior for other backends, but ZE memory flows were not covered in this area.
Adding ZE support and tests improves confidence that:
How?
The change adds ZE-gated code paths under HAVE_ZE in gtest mem_buffer:
A new UCM ZE hooks gtest suite allocates and frees ZE host, device, and managed memory and verifies that matching UCM alloc/free events are generated.
Build integration is updated in gtest/Makefile.am to include the new ZE test sources and ZE compiler/linker flags when ZE support is enabled.