UCM/ZE/TEST: Add ZE support to gtest mem_buffer and UCM hook tests - #11321
Merged
brminich merged 20 commits intoAug 6, 2026
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.
yafshar
marked this pull request as ready for review
April 4, 2026 14:57
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
May 12, 2026 18:46
01288fb to
072a8d0
Compare
Contributor
Author
tvegas1
reviewed
Jul 8, 2026
Remove the dead HAVE_ZE guard around ZE memset case labels because the enum values are always defined. Use UCS_TEST_ABORT instead of UCS_TEST_SKIP in ZE allocate() paths for consistency with memset and copy_between handling. Also drop the duplicate -lze_loader from gtest_LDADD since it is already included in ZE_LIBS. Signed-off-by: Yaser Afshar <yaser.afshar@intel.com>
tvegas1
approved these changes
Jul 8, 2026
Contributor
Author
|
@tvegas1 can we merge this PR? |
Contributor
Author
brminich
approved these changes
Aug 6, 2026
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
3 participants
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.