AI/rocm-systems
2
0
Forka 0
Codice Problemi Pull Request Actions Pacchetti Progetti Rilasci Wiki Attività

rocrtst/dtif: performance::memory_async_copy test fix on DTIF

Sfoglia il codice sorgente 
Signed-off-by: Aaron Liu <aaron.liu@amd.com>
Signed-off-by: Feifei Xu <feifxu@amd.com>
Signed-off-by: Longlong Yao <Longlong.Yao@amd.com>
Reviewed-by: David Yat Sin <David.YatSin@amd.com>


[ROCm/ROCR-Runtime commit: 297ea78140]
This commit is contained in:
Aaron Liu
2025-01-10 16:39:47 +08:00
committato da Yat Sin, David
parent b99015f30a
commit ba372ca4a8
1 ha cambiato i file con 14 aggiunte e 8 eliminazioni
Scarica il file Patch Scarica il file Diff Expand all files Collapse all files
projects/rocr-runtime/rocrtst/suites/performance/memory_async_copy.cc
+14 -8
Vedi File
@@ -69,6 +69,9 @@
} \
}
/* PCIE BDF ID: 0xC81407 is specific to DTIF platform */
static const uint32_t kDtifBdfId = 0xC81407;
constexpr const size_t MemoryAsyncCopy::Size[kNumGranularity];
constexpr const char* MemoryAsyncCopy::Str[kNumGranularity];
constexpr const int MemoryAsyncCopy::kMaxCopySize;
@@ -689,17 +692,20 @@ static hsa_status_t GetGPUAgents(hsa_agent_t agent, void* data) {
err = hsa_agent_get_info(agent, (hsa_agent_info_t)HSA_AMD_AGENT_INFO_DOMAIN, &pci_domain_id);
RET_IF_HSA_ERR(err);
hwloc_obj_t gpu_hwl_dev;
gpu_hwl_dev = hwloc_get_pcidev_by_busid(ptr->topology(), pci_domain_id, bus, device,
function);
hwloc_obj_t gpu_numa_node = nullptr;
if (agent_bdf_id != kDtifBdfId) {
hwloc_obj_t gpu_hwl_dev;
gpu_hwl_dev = hwloc_get_pcidev_by_busid(ptr->topology(), pci_domain_id, bus, device,
function);
if (gpu_hwl_dev == nullptr) {
return HSA_STATUS_ERROR;
if (gpu_hwl_dev == nullptr) {
return HSA_STATUS_ERROR;
}
gpu_numa_node = hwloc_get_ancestor_obj_by_type(ptr->topology(),
HWLOC_OBJ_NUMANODE, gpu_hwl_dev);
}
hwloc_obj_t gpu_numa_node = hwloc_get_ancestor_obj_by_type(ptr->topology(),
HWLOC_OBJ_NUMANODE, gpu_hwl_dev);
if (gpu_numa_node != nullptr) {
char s1[256], s2[256];
hwloc_bitmap_snprintf(s1, sizeof(s1), gpu_numa_node->nodeset);