mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-24 04:12:57 +08:00
38 lines
1.4 KiB
C++
38 lines
1.4 KiB
C++
|
|
/*
|
||
|
|
* Copyright (C) 2025 Intel Corporation
|
||
|
|
*
|
||
|
|
* SPDX-License-Identifier: MIT
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "shared/source/command_stream/host_function.h"
|
||
|
|
|
||
|
|
#include "shared/source/command_stream/command_stream_receiver.h"
|
||
|
|
#include "shared/source/command_stream/host_function_worker_atomic.h"
|
||
|
|
#include "shared/source/command_stream/host_function_worker_counting_semaphore.h"
|
||
|
|
#include "shared/source/command_stream/host_function_worker_cv.h"
|
||
|
|
#include "shared/source/command_stream/host_function_worker_interface.h"
|
||
|
|
|
||
|
|
namespace NEO::HostFunctionFactory {
|
||
|
|
|
||
|
|
IHostFunctionWorker *createHostFunctionWorker(int32_t hostFunctionWorkerMode,
|
||
|
|
bool isAubMode,
|
||
|
|
const std::function<void(GraphicsAllocation &)> &downloadAllocationImpl,
|
||
|
|
GraphicsAllocation *allocation,
|
||
|
|
HostFunctionData *data) {
|
||
|
|
|
||
|
|
bool skipHostFunctionExecution = isAubMode;
|
||
|
|
|
||
|
|
switch (hostFunctionWorkerMode) {
|
||
|
|
default:
|
||
|
|
case 0:
|
||
|
|
return new HostFunctionWorkerCountingSemaphore(skipHostFunctionExecution, downloadAllocationImpl, allocation, data);
|
||
|
|
case 1:
|
||
|
|
return new HostFunctionWorkerCV(skipHostFunctionExecution, downloadAllocationImpl, allocation, data);
|
||
|
|
case 2:
|
||
|
|
return new HostFunctionWorkerAtomic(skipHostFunctionExecution, downloadAllocationImpl, allocation, data);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace NEO::HostFunctionFactory
|