Disable two flaky tests to keep CI green (NVIDIA#4283)

wujingyue · web-flow · commit bb5b38cdadf4 · 2025-04-18T21:39:26.000-07:00
diff --git a/tests/cpp/test_multidevice_communications.cpp b/tests/cpp/test_multidevice_communications.cpp
@@ -417,7 +417,7 @@ INSTANTIATE_TEST_SUITE_P(
 
 using P2PCommunicationTest = MultiDeviceTest;
 
-TEST_F(P2PCommunicationTest, CudaComm) {
+TEST_F(P2PCommunicationTest, DISABLED_CudaComm) {
   static constexpr int kTensorSize = 8;
   static constexpr int kNumRepetitions = 32;
 
diff --git a/tests/cpp/test_multidevice_host_ir.cpp b/tests/cpp/test_multidevice_host_ir.cpp
@@ -478,7 +478,7 @@ TEST_F(OverlapDistributedMatmulTest, AG_linear) {
   EXPECT_TRUE(torch::allclose(out_ref, out_at, 1e-1, 1e-1));
 }
 
-TEST_F(MultiDeviceTest, ShareIpcMemHandles) {
+TEST_F(MultiDeviceTest, DISABLED_ShareIpcMemHandles) {
   static constexpr int kTensorSize = 4;
   static constexpr int kNumRepetitions = 10;
 

Original file line number	Diff line number	Diff line change
`@@ -478,7 +478,7 @@ TEST_F(OverlapDistributedMatmulTest, AG_linear) {`
`478`	`478`	`EXPECT_TRUE(torch::allclose(out_ref, out_at, 1e-1, 1e-1));`
`479`	`479`	`}`
`480`	`480`
`481`		`-TEST_F(MultiDeviceTest, ShareIpcMemHandles) {`
	`481`	`+TEST_F(MultiDeviceTest, DISABLED_ShareIpcMemHandles) {`
`482`	`482`	`static constexpr int kTensorSize = 4;`
`483`	`483`	`static constexpr int kNumRepetitions = 10;`
`484`	`484`