diff --git a/tests/kfdtest/src/KFDQMTest.cpp b/tests/kfdtest/src/KFDQMTest.cpp index 10f74ae808..f2b68ce233 100644 --- a/tests/kfdtest/src/KFDQMTest.cpp +++ b/tests/kfdtest/src/KFDQMTest.cpp @@ -1671,11 +1671,16 @@ TEST_F(KFDQMTest, P2PTest) { HsaMemFlags memFlags = {0}; HsaMemMapFlags mapFlags = {0}; memFlags.ui32.PageSize = HSA_PAGE_SIZE_4KB; - memFlags.ui32.HostAccess = 0; + memFlags.ui32.HostAccess = 1; memFlags.ui32.NonPaged = 1; memFlags.ui32.NoNUMABind = 1; unsigned int end = size / sizeof(HSAuint32) - 1; + if (!m_NodeInfo.IsGPUNodeLargeBar(g_TestDstNodeId) && + m_NodeInfo.AreGPUNodesXGMI(g_TestNodeId, g_TestDstNodeId)) { + memFlags.ui32.HostAccess = 0; + } + /* 1. Allocate a system buffer and allow the access to GPUs */ EXPECT_SUCCESS(hsaKmtAllocMemory(0, size, memFlags, reinterpret_cast(&sysBuf)));