mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-27 15:53:13 +08:00
370424a1e0
Change-Id: Iaec903af420f0a92f7d86e484c83300fb9c531ad Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
59 lines
2.2 KiB
C++
59 lines
2.2 KiB
C++
/*
|
|
* Copyright (C) 2018-2020 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "execution_environment/execution_environment.h"
|
|
#include "execution_environment/root_device_environment.h"
|
|
#include "os_interface/device_factory.h"
|
|
#include "opencl/source/command_stream/aub_command_stream_receiver.h"
|
|
#include "opencl/source/command_stream/command_stream_receiver_with_aub_dump.h"
|
|
#include "opencl/source/command_stream/tbx_command_stream_receiver.h"
|
|
|
|
namespace NEO {
|
|
|
|
extern CommandStreamReceiverCreateFunc commandStreamReceiverFactory[IGFX_MAX_CORE];
|
|
|
|
CommandStreamReceiver *createCommandStreamImpl(ExecutionEnvironment &executionEnvironment, uint32_t rootDeviceIndex) {
|
|
auto funcCreate = commandStreamReceiverFactory[executionEnvironment.rootDeviceEnvironments[rootDeviceIndex]->getHardwareInfo()->platform.eRenderCoreFamily];
|
|
if (funcCreate == nullptr) {
|
|
return nullptr;
|
|
}
|
|
CommandStreamReceiver *commandStreamReceiver = nullptr;
|
|
int32_t csr = DebugManager.flags.SetCommandStreamReceiver.get();
|
|
if (csr < 0) {
|
|
csr = CommandStreamReceiverType::CSR_HW;
|
|
}
|
|
switch (csr) {
|
|
case CSR_HW:
|
|
commandStreamReceiver = funcCreate(false, executionEnvironment, rootDeviceIndex);
|
|
break;
|
|
case CSR_AUB:
|
|
commandStreamReceiver = AUBCommandStreamReceiver::create("aubfile", true, executionEnvironment, rootDeviceIndex);
|
|
break;
|
|
case CSR_TBX:
|
|
commandStreamReceiver = TbxCommandStreamReceiver::create("", false, executionEnvironment, rootDeviceIndex);
|
|
break;
|
|
case CSR_HW_WITH_AUB:
|
|
commandStreamReceiver = funcCreate(true, executionEnvironment, rootDeviceIndex);
|
|
break;
|
|
case CSR_TBX_WITH_AUB:
|
|
commandStreamReceiver = TbxCommandStreamReceiver::create("aubfile", true, executionEnvironment, rootDeviceIndex);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
return commandStreamReceiver;
|
|
}
|
|
|
|
bool getDevicesImpl(size_t &numDevicesReturned, ExecutionEnvironment &executionEnvironment) {
|
|
if (DeviceFactory::isHwModeSelected()) {
|
|
return DeviceFactory::getDevices(numDevicesReturned, executionEnvironment);
|
|
}
|
|
return DeviceFactory::getDevicesForProductFamilyOverride(numDevicesReturned, executionEnvironment);
|
|
}
|
|
|
|
} // namespace NEO
|