mirror of
https://github.com/intel/compute-runtime.git
synced 2026-01-06 02:18:05 +08:00
Add preemption flags creation to shared code
Signed-off-by: Zbigniew Zdanowicz <zbigniew.zdanowicz@intel.com>
This commit is contained in:
committed by
Compute-Runtime-Automation
parent
3b35ba504f
commit
e295e74c0a
@@ -11,28 +11,20 @@
|
||||
|
||||
namespace NEO {
|
||||
|
||||
void ClPreemptionHelper::setPreemptionLevelFlags(PreemptionFlags &flags, Device &device, Kernel *kernel) {
|
||||
if (kernel) {
|
||||
const auto &kernelDescriptor = kernel->getKernelInfo().kernelDescriptor;
|
||||
flags.flags.disabledMidThreadPreemptionKernel = kernelDescriptor.kernelAttributes.flags.requiresDisabledMidThreadPreemption;
|
||||
flags.flags.vmeKernel = kernel->isVmeKernel();
|
||||
flags.flags.usesFencesForReadWriteImages = kernelDescriptor.kernelAttributes.flags.usesFencesForReadWriteImages;
|
||||
flags.flags.schedulerKernel = kernel->isSchedulerKernel;
|
||||
}
|
||||
flags.flags.deviceSupportsVmePreemption = device.getDeviceInfo().vmeAvcSupportsPreemption;
|
||||
flags.flags.disablePerCtxtPreemptionGranularityControl = device.getHardwareInfo().workaroundTable.waDisablePerCtxtPreemptionGranularityControl;
|
||||
flags.flags.disableLSQCROPERFforOCL = device.getHardwareInfo().workaroundTable.waDisableLSQCROPERFforOCL;
|
||||
}
|
||||
|
||||
PreemptionMode ClPreemptionHelper::taskPreemptionMode(Device &device, const MultiDispatchInfo &multiDispatchInfo) {
|
||||
PreemptionMode devMode = device.getPreemptionMode();
|
||||
|
||||
for (const auto &di : multiDispatchInfo) {
|
||||
auto kernel = di.getKernel();
|
||||
|
||||
PreemptionFlags flags = {};
|
||||
setPreemptionLevelFlags(flags, device, kernel);
|
||||
const KernelDescriptor *kernelDescriptor = nullptr;
|
||||
bool schedulerKernel = false;
|
||||
if (kernel != nullptr) {
|
||||
kernelDescriptor = &kernel->getDescriptor();
|
||||
schedulerKernel = kernel->isSchedulerKernel;
|
||||
}
|
||||
|
||||
PreemptionFlags flags = PreemptionHelper::createPreemptionLevelFlags(device, kernelDescriptor, schedulerKernel);
|
||||
PreemptionMode taskMode = PreemptionHelper::taskPreemptionMode(devMode, flags);
|
||||
if (devMode > taskMode) {
|
||||
devMode = taskMode;
|
||||
|
||||
Reference in New Issue
Block a user