From c1b720c5df5415741eb906f1130dbfe3b8c2415b Mon Sep 17 00:00:00 2001 From: Peter Andreas Entschev Date: Tue, 16 Jul 2024 08:33:34 -0700 Subject: [PATCH] Test RMM memory get/put with protov2 only --- ci/test_common.sh | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/ci/test_common.sh b/ci/test_common.sh index 06b7c60e..90849e4c 100755 --- a/ci/test_common.sh +++ b/ci/test_common.sh @@ -35,10 +35,17 @@ run_cpp_tests() { RUNTIME_PATH=${CONDA_PREFIX:-./} BINARY_PATH=${RUNTIME_PATH}/bin - CMD_LINE="timeout 10m ${BINARY_PATH}/gtests/libucxx/UCXX_TEST" + # Disable memory get/put with RMM in protov1, it always segfaults. + CMD_LINE="timeout 10m ${BINARY_PATH}/gtests/libucxx/UCXX_TEST --gtest_filter=-*RMM*Memory*" log_command "${CMD_LINE}" UCX_TCP_CM_REUSEADDR=y ${CMD_LINE} + + # Only test memory get/put with RMM in protov2, as protov1 segfaults. + CMD_LINE="timeout 10m ${BINARY_PATH}/gtests/libucxx/UCXX_TEST --gtest_filter=*RMM*Memory*" + + log_command "${CMD_LINE}" + UCX_PROTO_ENABLE=y UCX_TCP_CM_REUSEADDR=y ${CMD_LINE} } run_cpp_benchmark() {