From a206abcaa14ac3f7afce3e67844566965a5f8dfd Mon Sep 17 00:00:00 2001 From: Satyanvesh Dittakavi Date: Tue, 10 May 2022 17:43:40 +0000 Subject: [PATCH] SWDEV-335780 - Address OOM issue with BERT MLPerf Memory free was not happening due to no flush Change-Id: Iee9d2c9f8c50aa9b6012ca9d8fd792a51973b8c8 [ROCm/clr commit: f6e21144ae60e9a83b79966bfda3f0b115984d00] --- projects/clr/hipamd/src/hip_event.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/projects/clr/hipamd/src/hip_event.cpp b/projects/clr/hipamd/src/hip_event.cpp index b157e88350..3d6f4283e1 100644 --- a/projects/clr/hipamd/src/hip_event.cpp +++ b/projects/clr/hipamd/src/hip_event.cpp @@ -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; }