mirror of
https://github.com/intel/compute-runtime.git
synced 2025-09-15 13:01:45 +08:00

Before performing gpu device reset, first all level zero resources and gpu device specific resources have to be cleaned up. Also as after device reset, state of gpu device would be lost. Hence after performing gpu device reset, level zero device have to be reinitialized by querying gpu device again. This change is aimed at reinitializing the level zero resources after gpu device reset, so that user could continue using level zero devices after device reset. Related-To: LOCI-2627 Signed-off-by: Jitendra Sharma <jitendra.sharma@intel.com>
70 lines
2.9 KiB
C++
70 lines
2.9 KiB
C++
/*
|
|
* Copyright (C) 2018-2021 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/command_stream/aub_command_stream_receiver.h"
|
|
#include "shared/source/command_stream/command_stream_receiver_with_aub_dump.h"
|
|
#include "shared/source/command_stream/tbx_command_stream_receiver.h"
|
|
#include "shared/source/execution_environment/execution_environment.h"
|
|
#include "shared/source/execution_environment/root_device_environment.h"
|
|
#include "shared/source/helpers/api_specific_config.h"
|
|
#include "shared/source/os_interface/device_factory.h"
|
|
|
|
namespace NEO {
|
|
|
|
extern CommandStreamReceiverCreateFunc commandStreamReceiverFactory[2 * IGFX_MAX_CORE];
|
|
|
|
CommandStreamReceiver *createCommandStreamImpl(ExecutionEnvironment &executionEnvironment,
|
|
uint32_t rootDeviceIndex,
|
|
const DeviceBitfield deviceBitfield) {
|
|
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, deviceBitfield);
|
|
break;
|
|
case CSR_AUB:
|
|
commandStreamReceiver = AUBCommandStreamReceiver::create(ApiSpecificConfig::getName(), true, executionEnvironment, rootDeviceIndex, deviceBitfield);
|
|
break;
|
|
case CSR_TBX:
|
|
commandStreamReceiver = TbxCommandStreamReceiver::create("", false, executionEnvironment, rootDeviceIndex, deviceBitfield);
|
|
break;
|
|
case CSR_HW_WITH_AUB:
|
|
commandStreamReceiver = funcCreate(true, executionEnvironment, rootDeviceIndex, deviceBitfield);
|
|
break;
|
|
case CSR_TBX_WITH_AUB:
|
|
commandStreamReceiver = TbxCommandStreamReceiver::create(ApiSpecificConfig::getName(), true, executionEnvironment, rootDeviceIndex, deviceBitfield);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
return commandStreamReceiver;
|
|
}
|
|
|
|
bool prepareDeviceEnvironmentsImpl(ExecutionEnvironment &executionEnvironment) {
|
|
if (DeviceFactory::isHwModeSelected()) {
|
|
return DeviceFactory::prepareDeviceEnvironments(executionEnvironment);
|
|
}
|
|
return DeviceFactory::prepareDeviceEnvironmentsForProductFamilyOverride(executionEnvironment);
|
|
}
|
|
|
|
bool prepareDeviceEnvironmentImpl(ExecutionEnvironment &executionEnvironment, std::string &osPciPath, const uint32_t rootDeviceIndex) {
|
|
if (DeviceFactory::isHwModeSelected()) {
|
|
return DeviceFactory::prepareDeviceEnvironment(executionEnvironment, osPciPath, rootDeviceIndex);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
} // namespace NEO
|