AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-328349 - Rate limit hostcall listener

Browse Source 
Update timeout for hostcall wait for signal. If the timeout is small it
checks frequent enough to affect performance for certain applications
which may be CPU bound.

Change-Id: I0a879559e4ad111b09a994a5b82a6faf6e4fea3f


[ROCm/clr commit: 9292abb2d8]
This commit is contained in:
Saleel Kudchadker
2022-03-18 11:28:55 -07:00
parent e43702deeb
commit 0689be6c0e
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/clr/rocclr/device/devhostcall.cpp
+9 -1
View File
@@ -270,17 +270,25 @@ class HostcallListener {
HostcallListener* hostcallListener = nullptr;
amd::Monitor listenerLock("Hostcall listener lock");
constexpr static uint64_t kTimeoutFloor = K * K * 4;
constexpr static uint64_t kTimeoutCeil = K * K * 16;
void HostcallListener::consumePackets() {
uint64_t timeout = 1024 * 1024;
uint64_t timeout = kTimeoutFloor;
uint64_t signal_value = SIGNAL_INIT;
while (true) {
while (true) {
uint64_t new_value = doorbell_->Wait(signal_value, device::Signal::Condition::Ne, timeout);
if (new_value != signal_value) {
signal_value = new_value;
// Reduce the timeout for quicker processing
timeout = timeout >> 0x1;
timeout = std::max(kTimeoutFloor, timeout);
break;
}
// Increase the timeout since we dont need to check as frequently
timeout = timeout << 0x1;
timeout = std::min(kTimeoutCeil, timeout);
}
if (signal_value == SIGNAL_DONE) {