86 lines
2.4 KiB
C++
86 lines
2.4 KiB
C++
/*
|
|
* Copyright (C) 2017-2018 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "runtime/helpers/hw_info.h"
|
|
#include "runtime/helpers/options.h"
|
|
#include "runtime/os_interface/device_factory.h"
|
|
#include "runtime/os_interface/linux/drm_neo.h"
|
|
#include "runtime/os_interface/hw_info_config.h"
|
|
#include "runtime/os_interface/linux/os_interface.h"
|
|
#include "drm/i915_drm.h"
|
|
#include "runtime/gmm_helper/gmm_helper.h"
|
|
#include "runtime/device/device.h"
|
|
|
|
#include <vector>
|
|
#include <cstring>
|
|
|
|
namespace OCLRT {
|
|
|
|
size_t DeviceFactory::numDevices = 0;
|
|
HardwareInfo *DeviceFactory::hwInfos = nullptr;
|
|
|
|
bool DeviceFactory::getDevices(HardwareInfo **pHWInfos, size_t &numDevices, ExecutionEnvironment &executionEnvironment) {
|
|
std::vector<HardwareInfo> tHwInfos;
|
|
unsigned int devNum = 0;
|
|
size_t requiredDeviceCount = 1;
|
|
|
|
if (DebugManager.flags.CreateMultipleDevices.get()) {
|
|
requiredDeviceCount = DebugManager.flags.CreateMultipleDevices.get();
|
|
}
|
|
|
|
Drm *drm = Drm::create(devNum);
|
|
if (!drm) {
|
|
return false;
|
|
}
|
|
|
|
executionEnvironment.osInterface.reset(new OSInterface());
|
|
executionEnvironment.osInterface->get()->setDrm(drm);
|
|
|
|
const HardwareInfo *pCurrDevice = platformDevices[devNum];
|
|
|
|
while (devNum < requiredDeviceCount) {
|
|
HardwareInfo tmpHwInfo;
|
|
HwInfoConfig *hwConfig = HwInfoConfig::get(pCurrDevice->pPlatform->eProductFamily);
|
|
if (hwConfig->configureHwInfo(pCurrDevice, &tmpHwInfo, executionEnvironment.osInterface.get())) {
|
|
return false;
|
|
}
|
|
tHwInfos.push_back(tmpHwInfo);
|
|
devNum++;
|
|
}
|
|
|
|
HardwareInfo *ptr = new HardwareInfo[devNum];
|
|
for (size_t i = 0; i < tHwInfos.size(); i++)
|
|
ptr[i] = tHwInfos[i];
|
|
|
|
numDevices = devNum;
|
|
*pHWInfos = ptr;
|
|
|
|
DeviceFactory::numDevices = devNum;
|
|
DeviceFactory::hwInfos = ptr;
|
|
|
|
return true;
|
|
}
|
|
|
|
void DeviceFactory::releaseDevices() {
|
|
if (DeviceFactory::numDevices > 0) {
|
|
for (unsigned int i = 0; i < DeviceFactory::numDevices; ++i) {
|
|
Drm::closeDevice(i);
|
|
delete hwInfos[i].pSysInfo;
|
|
delete hwInfos[i].pSkuTable;
|
|
delete hwInfos[i].pWaTable;
|
|
delete hwInfos[i].pPlatform;
|
|
}
|
|
delete[] hwInfos;
|
|
}
|
|
DeviceFactory::hwInfos = nullptr;
|
|
DeviceFactory::numDevices = 0;
|
|
}
|
|
|
|
void Device::appendOSExtensions(std::string &deviceExtensions) {
|
|
}
|
|
} // namespace OCLRT
|