adjusting some tests

microsoft · Apr 15, 2024 · 249ad84 · 249ad84
1 parent db263e7
commit 249ad84
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/test/mp_unit/communicator_tests.cu b/test/mp_unit/communicator_tests.cu
@@ -101,7 +101,7 @@ void CommunicatorTest::SetUp() {
 
   ASSERT_EQ((deviceBufferSize / sizeof(int)) % gEnv->worldSize, 0);
 
-  connectMesh(false, false, true);
+  connectMesh(true, true, false);
 
   devicePtr.resize(numBuffers);
   localMemory.resize(numBuffers);
@@ -116,10 +116,10 @@ void CommunicatorTest::SetUp() {
 
   for (size_t n = 0; n < numBuffers; n++) {
     devicePtr[n] = mscclpp::allocSharedCuda<int>(deviceBufferSize / sizeof(int));
-    //registerMemoryPairs(devicePtr[n].get(), deviceBufferSize, mscclpp::Transport::CudaIpc | ibTransport, 0, remoteRanks,
-    //                    localMemory[n], remoteMemory[n]);
-    registerMemoryPairs(devicePtr[n].get(), deviceBufferSize, mscclpp::Transport::Ethernet, 0, remoteRanks,
-                        localMemory[n], remoteMemory[n]);               
+    registerMemoryPairs(devicePtr[n].get(), deviceBufferSize, mscclpp::Transport::CudaIpc | ibTransport, 0, remoteRanks,
+                        localMemory[n], remoteMemory[n]);
+    //registerMemoryPairs(devicePtr[n].get(), deviceBufferSize, mscclpp::Transport::Ethernet, 0, remoteRanks,
+    //                    localMemory[n], remoteMemory[n]);               
   }
 }
 
@@ -289,7 +289,7 @@ TEST_F(CommunicatorTest, WriteWithHostSemaphores) {
   communicator->bootstrap()->barrier();
 }
 
-TEST_F(CommunicatorTest, TestEthernetConnection) {
+/*TEST_F(CommunicatorTest, TestEthernetConnection) {
   if (gEnv->rank >= numRanksToUse) return;
 
   deviceBufferInit();
@@ -309,4 +309,4 @@ TEST_F(CommunicatorTest, TestEthernetConnection) {
     }
   } while (!ready);
   communicator->bootstrap()->barrier();
-}
+}*/
diff --git a/test/mp_unit/proxy_channel_tests.cu b/test/mp_unit/proxy_channel_tests.cu
@@ -219,7 +219,7 @@ void ProxyChannelOneToOneTest::testPingPongPerf(bool useIPC, bool useIB, bool us
 
   auto* testInfo = ::testing::UnitTest::GetInstance()->current_test_info();
   const std::string testName = std::string(testInfo->test_suite_name()) + "." + std::string(testInfo->name());
-  const int nTries = 1000000;
+  const int nTries = 1000;
 
   // Warm-up
   kernelProxyPingPong<<<1, 1024>>>(buff.get(), gEnv->rank, 1, waitWithPoll, nTries, ret.get());