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() {