mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-30 09:58:55 +08:00
073a40c14c
- add requirePerContextMemorySpace flag to ExecEnv Related-To: NEO-4637 Change-Id: I3968bedd9cfd1b1e41e79b63939b7a9bf25e7ff5 Signed-off-by: Mateusz Hoppe <mateusz.hoppe@intel.com>
93 lines
2.5 KiB
C++
93 lines
2.5 KiB
C++
/*
|
|
* Copyright (C) 2019-2020 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "level_zero/core/source/driver/driver.h"
|
|
|
|
#include "shared/source/device/device.h"
|
|
#include "shared/source/execution_environment/execution_environment.h"
|
|
#include "shared/source/os_interface/debug_env_reader.h"
|
|
#include "shared/source/os_interface/device_factory.h"
|
|
|
|
#include "level_zero/core/source/device/device.h"
|
|
#include "level_zero/core/source/driver/driver_handle.h"
|
|
#include "level_zero/core/source/driver/driver_imp.h"
|
|
|
|
#include <memory>
|
|
#include <thread>
|
|
|
|
namespace L0 {
|
|
|
|
_ze_driver_handle_t *GlobalDriverHandle;
|
|
uint32_t driverCount = 1;
|
|
|
|
void DriverImp::initialize(bool *result) {
|
|
*result = false;
|
|
|
|
NEO::EnvironmentVariableReader envReader;
|
|
L0EnvVariables envVariables = {};
|
|
envVariables.affinityMask =
|
|
envReader.getSetting("ZE_AFFINITY_MASK", std::string(""));
|
|
envVariables.programDebugging =
|
|
envReader.getSetting("ZET_ENABLE_PROGRAM_DEBUGGING", false);
|
|
|
|
auto executionEnvironment = new NEO::ExecutionEnvironment();
|
|
UNRECOVERABLE_IF(nullptr == executionEnvironment);
|
|
|
|
if (envVariables.programDebugging) {
|
|
executionEnvironment->setPerContextMemorySpace();
|
|
}
|
|
|
|
executionEnvironment->incRefInternal();
|
|
auto neoDevices = NEO::DeviceFactory::createDevices(*executionEnvironment);
|
|
executionEnvironment->decRefInternal();
|
|
if (!neoDevices.empty()) {
|
|
GlobalDriverHandle = DriverHandle::create(std::move(neoDevices), envVariables);
|
|
if (GlobalDriverHandle != nullptr) {
|
|
*result = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool DriverImp::initStatus(false);
|
|
|
|
ze_result_t DriverImp::driverInit(ze_init_flag_t flag) {
|
|
std::call_once(initDriverOnce, [this]() {
|
|
bool result;
|
|
this->initialize(&result);
|
|
initStatus = result;
|
|
});
|
|
return ((initStatus) ? ZE_RESULT_SUCCESS : ZE_RESULT_ERROR_UNINITIALIZED);
|
|
}
|
|
|
|
ze_result_t driverHandleGet(uint32_t *pCount, ze_driver_handle_t *phDriverHandles) {
|
|
if (*pCount == 0) {
|
|
*pCount = driverCount;
|
|
return ZE_RESULT_SUCCESS;
|
|
}
|
|
|
|
if (*pCount > driverCount) {
|
|
*pCount = driverCount;
|
|
}
|
|
|
|
if (phDriverHandles == nullptr) {
|
|
return ZE_RESULT_ERROR_INVALID_ARGUMENT;
|
|
}
|
|
|
|
for (uint32_t i = 0; i < *pCount; i++) {
|
|
phDriverHandles[i] = GlobalDriverHandle;
|
|
}
|
|
|
|
return ZE_RESULT_SUCCESS;
|
|
}
|
|
|
|
static DriverImp driverImp;
|
|
Driver *Driver::driver = &driverImp;
|
|
|
|
ze_result_t init(ze_init_flag_t flag) { return Driver::get()->driverInit(flag); }
|
|
|
|
} // namespace L0
|