Sync streams when freeing or destroying mem
Change-Id: I6932f225a8b932bb2adbd5e37880f7e604496809
[ROCm/clr commit: 68df8efe90]
Ta zmiana jest zawarta w:
%! (template.HTML=
Saleel Kudchadker)złożyło 2020-03-20 10:53:23 -07:00
1 rodzic
ee23705de2
złożenie
adc574b2ed
Plik 1 zmieniony
+14
-1
@@ -274,6 +274,13 @@ hipError_t hipHostFree(void* ptr) {
|
||||
HIP_INIT_API(hipHostFree, ptr);
|
||||
|
||||
if (amd::SvmBuffer::malloced(ptr)) {
|
||||
for (auto& dev : g_devices) {
|
||||
amd::HostQueue* queue = hip::getNullStream(*dev->asContext());
|
||||
if (queue != nullptr) {
|
||||
queue->finish();
|
||||
}
|
||||
hip::syncStreams(dev->deviceId());
|
||||
}
|
||||
amd::SvmBuffer::free(*hip::getCurrentDevice()->asContext(), ptr);
|
||||
HIP_RETURN(hipSuccess);
|
||||
}
|
||||
@@ -289,7 +296,13 @@ hipError_t ihipArrayDestroy(hipArray* array) {
|
||||
if (is_valid(memObj) == false) {
|
||||
return hipErrorInvalidValue;
|
||||
}
|
||||
|
||||
for (auto& dev : g_devices) {
|
||||
amd::HostQueue* queue = hip::getNullStream(*dev->asContext());
|
||||
if (queue != nullptr) {
|
||||
queue->finish();
|
||||
}
|
||||
hip::syncStreams(dev->deviceId());
|
||||
}
|
||||
as_amd(memObj)->release();
|
||||
|
||||
delete array;
|
||||
|
||||
Reference in new issue
Zablokuj użytkownika