Files
compute-runtime/shared/source/command_stream/host_function.cpp

38 lines
1.4 KiB
C++
Raw Normal View History

/*
* 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