AI/rocm-systems
2
0
Rozštěpit 0
Zdrojový kód Úkoly Pull requesty Akce Balíčky Projekty Vydání Wiki Aktivita

kfdtest: add xgmi path for p2p tests

Procházet zdrojové kódy 
When large bar is not available, we can use
xgmi to do p2p tests.

Change-Id: Ib7b59fb8a4d41f605739a0428973f6b2f1a3450f
Signed-off-by: Eric Huang <JinhuiEric.Huang@amd.com>
Tento commit je obsažen v:
Eric Huang
2019-10-04 16:49:38 -04:00
odevzdal JinHuiEric Huang
rodič 6933540c81
revize 0174377351
5 změnil soubory, kde provedl 75 přidání a 29 odebrání
Stáhněte soubor záplaty Stáhněte rozdílový soubor Rozbalit všechny soubory Sbalit všechny soubory
tests/kfdtest/src/KFDPerformanceTest.cpp
+20 -12
Zobrazit soubor
@@ -147,15 +147,24 @@ TEST_F(KFDPerformanceTest, P2PBandWidthTest) {
const std::vector<int> gpuNodes = m_NodeInfo.GetNodesWithGPU();
std::vector<HSAuint32> nodes;
const bool isSpecified = g_TestDstNodeId != -1 && g_TestNodeId != -1;
int numPeers = 0;
for (unsigned i = 0; i < gpuNodes.size(); i++)
if (m_NodeInfo.IsGPUNodeLargeBar(gpuNodes.at(i)) &&
/* Users can use "--node=gpu1 --dst_node=gpu2" to specify devices */
(!isSpecified || gpuNodes.at(i) == g_TestDstNodeId || gpuNodes.at(i) == g_TestNodeId))
nodes.push_back(gpuNodes.at(i));
if (isSpecified) {
if (g_TestNodeId != g_TestDstNodeId) {
nodes.push_back(g_TestNodeId);
nodes.push_back(g_TestDstNodeId);
if ((m_NodeInfo.IsGPUNodeLargeBar(g_TestNodeId) &&
m_NodeInfo.IsGPUNodeLargeBar(g_TestDstNodeId)) ||
m_NodeInfo.AreGPUNodesXGMI(g_TestNodeId, g_TestDstNodeId))
numPeers = 2;
}
} else {
HSAint32 defaultGPU = m_NodeInfo.HsaDefaultGPUNode();
numPeers = m_NodeInfo.FindAccessiblePeers(&nodes, defaultGPU, true);
}
if (nodes.size() < 2) {
LOG() << "Skipping test: Need at least two large bar GPU." << std::endl;
if (numPeers < 2) {
LOG() << "Skipping test: Need at least two large bar GPU or XGMI connected." << std::endl;
return;
}
@@ -302,12 +311,11 @@ TEST_F(KFDPerformanceTest, P2POverheadTest) {
const std::vector<int> gpuNodes = m_NodeInfo.GetNodesWithGPU();
std::vector<HSAuint32> nodes;
for (unsigned i = 0; i < gpuNodes.size(); i++)
if (m_NodeInfo.IsGPUNodeLargeBar(gpuNodes.at(i)))
nodes.push_back(gpuNodes.at(i));
HSAint32 defaultGPU = m_NodeInfo.HsaDefaultGPUNode();
int numPeers = m_NodeInfo.FindAccessiblePeers(&nodes, defaultGPU, true);
if (nodes.size() < 2) {
LOG() << "Skipping test: Need at least two large bar GPU." << std::endl;
if (numPeers < 2) {
LOG() << "Skipping test: Need at least two large bar GPU or XGMI connected." << std::endl;
return;
}