SWDEV-299940 - Query global memory available on device with HSA attribute HSA_AMD_AGENT_INFO_MEMORY_AVAIL

This reverts commit 06593a072f.

Reason for revert: ROCr query should now be usable in upcoming release.

Change-Id: I2207761ca6af5d585d090bae1af09eb9a8e9bad6


[ROCm/clr commit: a52f5bda8f]
Этот коммит содержится в:
Maneesh Gupta
2022-11-07 01:22:05 -05:00
родитель 988bd8e0ea
Коммит f04740b919
+16 -3
Просмотреть файл
@@ -1669,10 +1669,23 @@ device::VirtualDevice* Device::createVirtualDevice(amd::CommandQueue* queue) {
bool Device::globalFreeMemory(size_t* freeMemory) const {
const uint TotalFreeMemory = 0;
const uint LargestFreeBlock = 1;
uint64_t globalAvailMemory;
// Queries memory available in bytes across all global pools owned by the agent
if (HSA_STATUS_SUCCESS !=
hsa_agent_get_info(bkendDevice_,
static_cast<hsa_agent_info_t>(HSA_AMD_AGENT_INFO_MEMORY_AVAIL),
&globalAvailMemory)) {
LogError("HSA_AMD_AGENT_INFO_MEMORY_AVAIL query failed.");
}
freeMemory[TotalFreeMemory] = freeMem_ / Ki;
freeMemory[TotalFreeMemory] -= (freeMemory[TotalFreeMemory] > HIP_HIDDEN_FREE_MEM * Ki) ?
HIP_HIDDEN_FREE_MEM * Ki : 0;
globalAvailMemory = globalAvailMemory / Ki;
if (globalAvailMemory > HIP_HIDDEN_FREE_MEM * Ki) {
globalAvailMemory -= HIP_HIDDEN_FREE_MEM * Ki;
} else {
globalAvailMemory = 0;
}
freeMemory[TotalFreeMemory] = globalAvailMemory;
// since there is no memory heap on ROCm, the biggest free block is
// equal to total free local memory
freeMemory[LargestFreeBlock] = freeMemory[TotalFreeMemory];