UCT/ZE/TEST: expand ZE RMA coverage and fix ze_copy rkey_compare path#11331
Open
yafshar wants to merge 6 commits intoopenucx:masterfrom
Open
UCT/ZE/TEST: expand ZE RMA coverage and fix ze_copy rkey_compare path#11331yafshar wants to merge 6 commits intoopenucx:masterfrom
yafshar wants to merge 6 commits intoopenucx:masterfrom
Conversation
- Add UCT RMA tests for ze_copy covering put/get zcopy operations on ZE device and managed memory. - Enable runtime skip logic based on ZE memory type availability instead of compile-time gtest filtering. - Include ze_copy in p2p_rma source memory-type selection alongside cuda_ipc and rocm_copy. - Introduce UCT_INSTANTIATE_ZE_TEST_CASE macro in uct_test.h to standardize ZE test instantiation. - Update gtest Makefile.am to build ZE transport tests.
- Wire ze_copy component rkey_compare to uct_base_rkey_compare to prevent crashes in ze_copy/test_md.rkey_compare. - Validate ze_copy mem_dereg parameters (memh and invalidate flag) to match MD bad-argument behavior when parameter checking is enabled. - Fix the style
- Add ZE host, device, and managed memory RMA tests for ze_copy transport in test_ze.cc. - Extend MD tests to cover ZE mem_query behavior and add ZE dmabuf query tests in test_md.cc.
yosefe
reviewed
Apr 12, 2026
Run dmabuf mem_query tests via the generic MD fixture (test_md_dmabuf) instead of a ZE-only fixture, validating the generic dmabuf interface across MDs.
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
2 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?
This PR expands ZE test coverage in UCT gtests and fixes a ZE MD callback issue
that caused a crash in existing tests.
in ze_copy_md.c
Why?
ZE paths lacked sufficient automated coverage for ze_copy transport and MD
behavior, and an existing ZE MD test exposed a real failure: