SWDEV-335780 - Address OOM issue with BERT MLPerf
Memory free was not happening due to no flush
Change-Id: Iee9d2c9f8c50aa9b6012ca9d8fd792a51973b8c8
[ROCm/clr commit: f6e21144ae]
Этот коммит содержится в:
@@ -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;
|
||||
}
|
||||
|
||||
Ссылка в новой задаче
Block a user