AI/rocm-systems
2
0 Vork 0
Code Kwesties Pull-aanvragen Actions Paketten Projecten Publicaties Wiki Activiteit

SWDEV-366636 - Fix performance drop in TF-RCCL models

Bladeren bron 
Change-Id: Idc845bb0dab858b94b9d2720cae8308cac2e7328
This commit is contained in:
Anusha GodavarthySurya authored and Anusha Godavarthy Surya committed 2023-02-17 00:48:13 -05:00
1 bovenliggende 6b873b8a86
commit f98dcf9d7f
4 files changed
+27 -8
Download Patch-bestand Download Diff-bestand Expand all files Collapse all files

No files matched your search

rocclr/device/rocm/rocdevice.cpp
+14 -2
Bestand weergeven
@@ -2710,10 +2710,22 @@ bool Device::SetClockMode(const cl_set_device_clock_mode_input_amd setClockModeI
return result;
}
// ================================================================================================
bool Device::IsHwEventReadyForcedWait(const amd::Event& event) const {
void* hw_event =
(event.NotifyEvent() != nullptr) ? event.NotifyEvent()->HwEvent() : event.HwEvent();
if (hw_event == nullptr) {
ClPrint(amd::LOG_INFO, amd::LOG_SIG, "No HW event");
return false;
}
static constexpr bool Timeout = true;
return WaitForSignal<Timeout>(reinterpret_cast<ProfilingSignal*>(hw_event)->signal_, false, true);
}
// ================================================================================================
bool Device::IsHwEventReady(const amd::Event& event, bool wait) const {
void* hw_event = (event.NotifyEvent() != nullptr) ?
event.NotifyEvent()->HwEvent() : event.HwEvent();
void* hw_event =
(event.NotifyEvent() != nullptr) ? event.NotifyEvent()->HwEvent() : event.HwEvent();
if (hw_event == nullptr) {
ClPrint(amd::LOG_INFO, amd::LOG_SIG, "No HW event");
return false;