From 7c62a1291827872bfee1f1e9991f887aa74bb4cf Mon Sep 17 00:00:00 2001
From: Eric Huang <jinhuieric.huang@amd.com>
Date: Mon, 6 Dec 2021 12:03:37 -0500
Subject: [PATCH] kfdtest: dynamically increase timeout for P2PBandWidthTest

Incease more timeout according to peers number to pass the
test on some PCIe link platforms.

Change-Id: Ifcb8c7297d6960c96fc18d29bc0a48733ca50165
Signed-off-by: Eric Huang <jinhuieric.huang@amd.com>
---
 tests/kfdtest/src/KFDPerformanceTest.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/kfdtest/src/KFDPerformanceTest.cpp b/tests/kfdtest/src/KFDPerformanceTest.cpp
index 32f342bd4d..bdd548c8e9 100644
--- a/tests/kfdtest/src/KFDPerformanceTest.cpp
+++ b/tests/kfdtest/src/KFDPerformanceTest.cpp
@@ -167,6 +167,8 @@ TEST_F(KFDPerformanceTest, P2PBandWidthTest) {
         return;
     }
 
+    g_TestTimeOut *= numPeers;
+
     std::vector<int> sysNodes(nodes); // include sysMem node 0...
     sysNodes.insert(sysNodes.begin(),0);
 
@@ -308,6 +310,8 @@ TEST_F(KFDPerformanceTest, P2PBandWidthTest) {
                                         (float)speed2 / 1024 << " GB/s" << std::endl;
         }
     }
+
+    g_TestTimeOut /= numPeers;
 exit:
     /* New line.*/
     LOG() << std::endl << msg.str() << std::endl;