mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-30 09:58:55 +08:00
a2fe929f0c
Cleaned up files: shared/source/command_stream/command_stream_receiver_hw.h shared/source/compiler_interface/compiler_interface.h shared/source/direct_submission/direct_submission_hw.h shared/source/helpers/dirty_state_helpers.h Related-To: NEO-5548 Signed-off-by: Warchulski, Jaroslaw <jaroslaw.warchulski@intel.com>
38 lines
1.2 KiB
C++
38 lines
1.2 KiB
C++
/*
|
|
* Copyright (C) 2021-2023 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "opencl/source/helpers/cl_preemption_helper.h"
|
|
|
|
#include "shared/source/compiler_interface/compiler_cache.h"
|
|
|
|
#include "opencl/source/helpers/dispatch_info.h"
|
|
|
|
namespace NEO {
|
|
|
|
PreemptionMode ClPreemptionHelper::taskPreemptionMode(Device &device, const MultiDispatchInfo &multiDispatchInfo) {
|
|
PreemptionMode devMode = device.getPreemptionMode();
|
|
|
|
for (const auto &di : multiDispatchInfo) {
|
|
auto kernel = di.getKernel();
|
|
|
|
const KernelDescriptor *kernelDescriptor = nullptr;
|
|
if (kernel != nullptr) {
|
|
kernelDescriptor = &kernel->getDescriptor();
|
|
}
|
|
|
|
PreemptionFlags flags = PreemptionHelper::createPreemptionLevelFlags(device, kernelDescriptor);
|
|
PreemptionMode taskMode = PreemptionHelper::taskPreemptionMode(devMode, flags);
|
|
if (devMode > taskMode) {
|
|
devMode = taskMode;
|
|
}
|
|
PRINT_DEBUG_STRING(DebugManager.flags.PrintDebugMessages.get(), stdout, "devMode = %d, taskMode = %d.\n",
|
|
static_cast<int>(device.getPreemptionMode()), static_cast<int>(taskMode));
|
|
}
|
|
return devMode;
|
|
}
|
|
} // namespace NEO
|