TEST/PROTO: Decrease TCP buffer size

openucx · Dec 19, 2023 · 20f3869 · 20f3869
1 parent e6acab0
commit 20f3869
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/test/gtest/ucp/test_ucp_request.cc b/test/gtest/ucp/test_ucp_request.cc
@@ -113,7 +113,7 @@ class test_proto_reset : public ucp_test {
         AM
     };
 
-    test_proto_reset() : m_msg_size(UCS_KBYTE * 128), m_completed(0)
+    test_proto_reset() : m_msg_size(UCS_KBYTE * 64), m_completed(0)
     {
         m_pending.head  = NULL;
         m_pending.ptail = NULL;
@@ -125,6 +125,8 @@ class test_proto_reset : public ucp_test {
             UCS_TEST_SKIP_R("reset is not supported for proto v1");
         }
 
+        ucs::scoped_setenv snd_buf("UCX_TCP_SNDBUF", "8K");
+        ucs::scoped_setenv ib_queue_len("UCX_IB_TX_QUEUE_LEN", "65");
         ucp_test::init();
         connect();
     }
@@ -283,6 +285,7 @@ class test_proto_reset : public ucp_test {
             rreq = ucp_tag_recv_nbx(receiver().worker(), rbuf.data(),
                                     m_msg_size, 0, 0, &param);
             ASSERT_TRUE(UCS_PTR_IS_PTR(rreq));
+            reqs.push_back(rreq);
             break;
 
         case RMA_GET:
@@ -314,13 +317,12 @@ class test_proto_reset : public ucp_test {
             ASSERT_TRUE(UCS_PTR_IS_PTR(sreq));
         }
 
-        reqs.push_back(rreq);
         reqs.push_back(sreq);
     }
 
     void reset_protocol(operation_e op, bool sync = false)
     {
-        static const unsigned reqs_count = 2000;
+        static const unsigned reqs_count = 1000;
         std::vector<void*> reqs;
         m_sbufs.resize(reqs_count);
         m_rbufs.resize(reqs_count);
@@ -469,8 +471,8 @@ UCS_TEST_P(test_proto_reset, stream_multi_zcopy_to_bcopy, "ZCOPY_THRESH=0",
     reset_protocol(STREAM);
 }
 
-UCS_TEST_SKIP_COND_P(test_proto_reset, rndv_am_zcopy_to_bcopy, has_transport("tcp"),
-                    "ZCOPY_THRESH=0", "RNDV_THRESH=0", "RNDV_SCHEME=am")
+UCS_TEST_P(test_proto_reset, rndv_am_zcopy_to_bcopy, "ZCOPY_THRESH=0",
+           "RNDV_THRESH=0", "RNDV_SCHEME=am")
 {
     reset_protocol(TAG);
 }