From f63c27446cf66b4bd91f18ac99fb123c71d7407d Mon Sep 17 00:00:00 2001 From: dmitrygx Date: Wed, 25 Nov 2020 12:17:34 +0200 Subject: [PATCH] GTEST/UCP: Expect SW RMA when managed memory is used --- test/gtest/ucp/test_ucp_mmap.cc | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/test/gtest/ucp/test_ucp_mmap.cc b/test/gtest/ucp/test_ucp_mmap.cc index 59987597a8f..d60a4754865 100644 --- a/test/gtest/ucp/test_ucp_mmap.cc +++ b/test/gtest/ucp/test_ucp_mmap.cc @@ -283,7 +283,10 @@ UCS_TEST_P(test_ucp_mmap, reg_mem_type) { ASSERT_UCS_OK(status); is_dummy = (size == 0); - test_rkey_management(&sender(), memh, is_dummy, is_tl_rdma()); + test_rkey_management(&sender(), memh, is_dummy, + is_tl_rdma() && + !UCP_MEM_IS_CUDA_MANAGED(alloc_mem_type) && + !UCP_MEM_IS_ROCM_MANAGED(alloc_mem_type)); status = ucp_mem_unmap(sender().ucph(), memh); ASSERT_UCS_OK(status);