intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2026-01-03 06:49:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: correct pollForCompletion in aub on program termination

Browse Source 
Related-To: NEO-14867

Signed-off-by: Grochowski, Stanislaw <stanislaw.grochowski@intel.com>
This commit is contained in:
Grochowski, Stanislaw
2025-07-10 13:41:47 +00:00
committed by Compute-Runtime-Automation
parent d3564fab68
commit e652679b95
7 changed files with 108 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
shared/source/device/device.cpp
View File

@@ -1085,10 +1085,20 @@ void Device::stopDirectSubmissionAndWaitForCompletion() {
}
void Device::pollForCompletion() {
if (allEngines.size() == 0 || !getDefaultEngine().commandStreamReceiver->isAubMode()) {
return;
}
for (auto &engine : allEngines) {
auto csr = engine.commandStreamReceiver;
csr->pollForCompletion();
}
for (auto &subDevice : subdevices) {
if (subDevice != nullptr) {
subDevice->pollForCompletion();
}
}
}
bool Device::isAnyDirectSubmissionEnabled() const {

2
shared/source/device/device.h
View File

@@ -212,7 +212,7 @@ class Device : public ReferenceTrackedObject<Device>, NEO::NonCopyableAndNonMova
return usmMemAllocPool.get();
}
MOCKABLE_VIRTUAL void stopDirectSubmissionAndWaitForCompletion();
void pollForCompletion();
MOCKABLE_VIRTUAL void pollForCompletion();
bool isAnyDirectSubmissionEnabled() const;
bool isAnyDirectSubmissionLightEnabled() const;
bool isStateSipRequired() const {