SWDEV-335780 - Address OOM issue with BERT MLPerf

Memory free was not happening due to no flush

Change-Id: Iee9d2c9f8c50aa9b6012ca9d8fd792a51973b8c8


[ROCm/clr commit: f6e21144ae]
Этот коммит содержится в:
Satyanvesh Dittakavi
2022-05-10 17:43:40 +00:00
родитель a7d73a753d
Коммит a206abcaa1
+3 -1
Просмотреть файл
@@ -215,15 +215,17 @@ hipError_t Event::recordCommand(amd::Command*& command, amd::HostQueue* queue,
if (command == nullptr) {
int32_t releaseFlags = ((ext_flags == 0) ? flags : ext_flags) &
(hipEventReleaseToSystem | hipEventReleaseToDevice);
bool markerTs = true;
if (releaseFlags & hipEventReleaseToDevice) {
releaseFlags = amd::Device::kCacheStateAgent;
} else if (releaseFlags & hipEventReleaseToSystem) {
releaseFlags = amd::Device::kCacheStateSystem;
} else {
releaseFlags = amd::Device::kCacheStateIgnore;
markerTs = false;
}
// Always submit a EventMarker.
command = new hip::EventMarker(*queue, !kMarkerDisableFlush, true, releaseFlags);
command = new hip::EventMarker(*queue, !kMarkerDisableFlush, markerTs, releaseFlags);
}
return hipSuccess;
}