Skip to content

Commit a03be17

Browse files
committed
UCP/TEST: separate ipc test case from others
1 parent 57c5686 commit a03be17

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

test/gtest/ucp/test_ucp_peer_failure.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -993,3 +993,4 @@ UCS_TEST_P(test_ucp_peer_failure_rndv_put_ppln_abort, pipeline,
993993
}
994994

995995
UCP_INSTANTIATE_TEST_CASE_GPU_AWARE(test_ucp_peer_failure_rndv_put_ppln_abort);
996+
UCP_INSTANTIATE_TEST_CASE_GPU_IPC(test_ucp_peer_failure_rndv_put_ppln_abort);

test/gtest/ucp/ucp_test.h

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -482,11 +482,19 @@ std::vector<ucp_test_param> enum_test_params(const std::string& tls)
482482
"srd") \
483483
UCP_INSTANTIATE_TEST_CASE_TLS_GPU_AWARE(_test_case, shm_ib, \
484484
"shm,ib,gdr_copy") \
485-
UCP_INSTANTIATE_TEST_CASE_TLS_GPU_AWARE(_test_case, shm_ib_ipc, \
486-
"shm,ib,cuda_ipc,rocm_ipc") \
487485
UCP_INSTANTIATE_TEST_CASE_TLS_GPU_AWARE(_test_case, ugni, \
488486
"ugni") \
489487
UCP_INSTANTIATE_TEST_CASE_TLS_GPU_AWARE(_test_case, tcp, \
490488
"tcp")
491489

490+
491+
/**
492+
* Instantiate the parameterized test case for gpu ipc transports
493+
*
494+
* @param _test_case Test case class, derived from ucp_test.
495+
*/
496+
#define UCP_INSTANTIATE_TEST_CASE_GPU_IPC(_test_case) \
497+
UCP_INSTANTIATE_TEST_CASE_TLS_GPU_AWARE(_test_case, shm_ib_ipc, \
498+
"shm,ib,cuda_ipc,rocm_ipc")
499+
492500
#endif

0 commit comments

Comments
 (0)