From 8ead739eba4d81b7706c0a2c6ca6514f150117e9 Mon Sep 17 00:00:00 2001 From: shizhibao Date: Fri, 20 Nov 2020 17:49:17 +0800 Subject: [PATCH] fix ucg_builtin_step_select_callbacks --- test/gtest/ucg/test_cb.cc | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/test/gtest/ucg/test_cb.cc b/test/gtest/ucg/test_cb.cc index 7a8497fa54d..6884962bd27 100644 --- a/test/gtest/ucg/test_cb.cc +++ b/test/gtest/ucg/test_cb.cc @@ -570,11 +570,11 @@ TEST_F(ucg_cb_test, test_step_callback_select_termonal) { int nonzero_length = 0; int flags = 0; - ucs_status_t ret = ucg_builtin_step_select_callbacks(phase, recv_cb, nonzero_length, flags); + ucs_status_t ret = ucg_builtin_step_select_callbacks(phase, 1, recv_cb, nonzero_length, flags); ASSERT_EQ(UCS_OK, ret); flags |= UCG_BUILTIN_OP_STEP_FLAG_SINGLE_ENDPOINT; - ret = ucg_builtin_step_select_callbacks(phase, recv_cb, nonzero_length, flags); + ret = ucg_builtin_step_select_callbacks(phase, 1, recv_cb, nonzero_length, flags); ASSERT_EQ(UCS_OK, ret); } } @@ -590,15 +590,15 @@ TEST_F(ucg_cb_test, test_step_callback_select_waypoint_fanout) { int nonzero_length = 0; int flags = 0; - ucs_status_t ret = ucg_builtin_step_select_callbacks(phase, recv_cb, nonzero_length, flags); + ucs_status_t ret = ucg_builtin_step_select_callbacks(phase, 1, recv_cb, nonzero_length, flags); ASSERT_EQ(UCS_OK, ret); nonzero_length = 1; - ret = ucg_builtin_step_select_callbacks(phase, recv_cb, nonzero_length, flags); + ret = ucg_builtin_step_select_callbacks(phase, 1, recv_cb, nonzero_length, flags); ASSERT_EQ(UCS_OK, ret); flags |= UCG_BUILTIN_OP_STEP_FLAG_FRAGMENTED; - ret = ucg_builtin_step_select_callbacks(phase, recv_cb, nonzero_length, flags); + ret = ucg_builtin_step_select_callbacks(phase, 1, recv_cb, nonzero_length, flags); ASSERT_EQ(UCS_OK, ret); } } @@ -614,11 +614,11 @@ TEST_F(ucg_cb_test, test_step_callback_select_reduce) { for (int nonzero_length = 0; nonzero_length < 2; nonzero_length++) { int flags = 0; - ucs_status_t ret = ucg_builtin_step_select_callbacks(phase, recv_cb, nonzero_length, flags); + ucs_status_t ret = ucg_builtin_step_select_callbacks(phase, 1, recv_cb, nonzero_length, flags); ASSERT_EQ(UCS_OK, ret); flags |= UCG_BUILTIN_OP_STEP_FLAG_SINGLE_ENDPOINT; - ret = ucg_builtin_step_select_callbacks(phase, recv_cb, nonzero_length, flags); + ret = ucg_builtin_step_select_callbacks(phase, 1, recv_cb, nonzero_length, flags); ASSERT_EQ(UCS_OK, ret); } } @@ -635,7 +635,7 @@ TEST_F(ucg_cb_test, test_step_callback_select_nonzero) { for (int nonzero_length = 0; nonzero_length < 2; nonzero_length++) { int flags = 0; - ucs_status_t ret = ucg_builtin_step_select_callbacks(phase, recv_cb, nonzero_length, flags); + ucs_status_t ret = ucg_builtin_step_select_callbacks(phase, 1, recv_cb, nonzero_length, flags); ASSERT_EQ(UCS_OK, ret); } } @@ -647,11 +647,11 @@ TEST_F(ucg_cb_test, test_step_callback_select_barrier) { int nonzero_length = 0; int flags = UCG_BUILTIN_OP_STEP_FLAG_LAST_STEP; - ucs_status_t ret = ucg_builtin_step_select_callbacks(phase, recv_cb, nonzero_length, flags); + ucs_status_t ret = ucg_builtin_step_select_callbacks(phase, 1, recv_cb, nonzero_length, flags); ASSERT_EQ(UCS_OK, ret); flags |= UCG_BUILTIN_OP_STEP_FLAG_SINGLE_ENDPOINT; - ret = ucg_builtin_step_select_callbacks(phase, recv_cb, nonzero_length, flags); + ret = ucg_builtin_step_select_callbacks(phase, 1, recv_cb, nonzero_length, flags); ASSERT_EQ(UCS_OK, ret); }