UCP/TEST: Revert redundant changes

openucx · May 15, 2024 · 2346a5b · 2346a5b
1 parent 7ee79f5
commit 2346a5b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/gtest/ucp/test_ucp_tag_xfer.cc b/test/gtest/ucp/test_ucp_tag_xfer.cc
@@ -1247,19 +1247,19 @@ UCS_TEST_P(multi_rail_max, max_lanes, "IB_NUM_PATHS?=64", "TM_SW_RNDV=y",
 
     ucp_lane_index_t num_lanes = ucp_ep_num_lanes(sender().ep());
     ASSERT_EQ(ucp_ep_num_lanes(receiver().ep()), num_lanes);
-    ASSERT_GE(num_lanes, max_lanes);
+    ASSERT_EQ(num_lanes, max_lanes);
 
-    size_t chunk_size = get_msg_size() / max_lanes;
+    size_t chunk_size = get_msg_size() / num_lanes;
 
-    for (ucp_lane_index_t lane = 0; lane < max_lanes; ++lane) {
+    for (ucp_lane_index_t lane = 0; lane < num_lanes; ++lane) {
         size_t sender_tx   = get_bytes_sent(sender().ep(), lane);
         size_t receiver_tx = get_bytes_sent(receiver().ep(), lane);
         UCS_TEST_MESSAGE << "lane[" << static_cast<int>(lane) << "] : "
                          << "sender " << sender_tx << " receiver " << receiver_tx;
 
         /* Verify that each lane sent something, except the active message lane
            that could be used only for control messages */
-        if (lane == max_lanes - 1) {
+        if (lane == num_lanes - 1) {
             EXPECT_GT(sender_tx + receiver_tx, 0); // last lane sends the rest
         } else if (lane != ucp_ep_get_am_lane(sender().ep())) {
             EXPECT_GE(sender_tx + receiver_tx, chunk_size);