GTEST/UCT: Relaxed test_cuda_nvml.device_get_fabric_info condition.

rakhmets · rakhmets · commit 35459fe57401 · 2025-08-27T12:27:28.000+03:00
diff --git a/test/gtest/uct/cuda/test_cuda_nvml.cc b/test/gtest/uct/cuda/test_cuda_nvml.cc
@@ -41,6 +41,6 @@ UCS_TEST_F(test_cuda_nvml, device_get_fabric_info) {
     fabric_info.version = nvmlGpuFabricInfo_v2;
     status = UCT_CUDA_NVML_WRAP_CALL(nvmlDeviceGetGpuFabricInfoV, device,
                                      &fabric_info);
-    EXPECT_EQ(status, UCS_OK);
+    EXPECT_TRUE((status == UCS_OK) || (status == UCS_ERR_IO_ERROR));
 }
 #endif

Original file line number	Diff line number	Diff line change
`@@ -41,6 +41,6 @@ UCS_TEST_F(test_cuda_nvml, device_get_fabric_info) {`
`41`	`41`	`fabric_info.version = nvmlGpuFabricInfo_v2;`
`42`	`42`	`status = UCT_CUDA_NVML_WRAP_CALL(nvmlDeviceGetGpuFabricInfoV, device,`
`43`	`43`	`&fabric_info);`
`44`		`- EXPECT_EQ(status, UCS_OK);`
	`44`	`+ EXPECT_TRUE((status == UCS_OK) \|\| (status == UCS_ERR_IO_ERROR));`
`45`	`45`	`}`
`46`	`46`	`#endif`