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

SWDEV-533220 - Release marker when HostQueue is destroyed (#460)

Browse Source 
Co-authored-by: Anusha GodavarthySurya <Anusha.GodavarthySurya@amd.com>
This commit is contained in:
Stojiljkovic, Vladana
2025-08-13 15:15:31 +02:00
committed by GitHub
parent 499e2ee044
commit 14760c6eba
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rocclr/platform/commandqueue.cpp
+9 -8
View File
@@ -72,20 +72,21 @@ bool HostQueue::terminate() {
if (lastCommand != nullptr) {
// Check if CPU batch wasn't flushed for completion with the last command
if (GetSubmissionBatch() != nullptr) {
auto command = new Marker(*this, false);
if (command != nullptr) {
ClPrint(LOG_DEBUG, LOG_CMD, "Marker queued to ensure finish");
command->enqueue();
lastCommand = command;
}
auto command = new Marker(*this, false);
if (command != nullptr) {
ClPrint(LOG_DEBUG, LOG_CMD, "Marker queued to ensure finish");
command->enqueue();
lastCommand->release();
lastCommand = command;
}
}
if (device_.gpu_error_ == CL_SUCCESS) {
lastCommand->awaitCompletion();
lastCommand->awaitCompletion();
}
// Note that if lastCommand isn't a marker, it may not be lastEnqueueCommand_ now
// after lastCommand->awaitCompletion() is called.
if (lastEnqueueCommand_ != nullptr) {
lastEnqueueCommand_ ->release(); // lastEnqueueCommand_ should be a marker
lastEnqueueCommand_->release(); // lastEnqueueCommand_ should be a marker
lastEnqueueCommand_ = nullptr;
}
lastCommand->release();