AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-532868 - Disable implicit wait in hipFree for async allocations (#323)

Browse Source 
* SWDEV-532868 - Disable implicit wait in hipFree for async allocations

* Fix compilation error
This commit is contained in:
Andryeyev, German
2025-05-21 11:44:38 -04:00
committed by GitHub
parent 7a357800dd
commit 58df22546b
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_memory.cpp
+3 -2
View File
@@ -106,12 +106,13 @@ hipError_t ihipFree(void *ptr) {
size_t offset = 0;
amd::Memory* memory_object = getMemoryObject(ptr, offset);
if (memory_object != nullptr) {
// Wait on the device, associated with the current memory object during allocation
auto device_id = memory_object->getUserData().deviceId;
g_devices[device_id]->SyncAllStreams();
// Find out if memory belongs to any memory pool
if (!g_devices[device_id]->FreeMemory(memory_object, nullptr)) {
// Wait on the device, associated with the current memory object during allocation
g_devices[device_id]->SyncAllStreams();
// External mem is not svm.
if (memory_object->isInterop()) {
amd::MemObjMap::RemoveMemObj(ptr);