mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-22 01:48:50 +08:00
058c30c9a8
- Make it avaialble also to OpenCL - Use it before parsing affinity mask, so devices are masked also following PCI order. Signed-off-by: Jaime Arteaga <jaime.a.arteaga.molina@intel.com>
43 lines
1.4 KiB
C++
43 lines
1.4 KiB
C++
/*
|
|
* Copyright (C) 2021 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/debug_settings/debug_settings_manager.h"
|
|
#include "shared/source/execution_environment/execution_environment.h"
|
|
#include "shared/source/execution_environment/root_device_environment.h"
|
|
#include "shared/source/os_interface/os_interface.h"
|
|
#include "shared/source/os_interface/windows/wddm/wddm.h"
|
|
|
|
#include <sstream>
|
|
|
|
namespace NEO {
|
|
|
|
bool comparePciIdBusNumber(std::unique_ptr<RootDeviceEnvironment> &rootDeviceEnvironment1, std::unique_ptr<RootDeviceEnvironment> &rootDeviceEnvironment2) {
|
|
// BDF sample format is : 00:02.0
|
|
auto bdfDevice1 = rootDeviceEnvironment1.get()->osInterface->getDriverModel()->as<NEO::Wddm>()->getAdapterBDF();
|
|
|
|
auto bdfDevice2 = rootDeviceEnvironment2.get()->osInterface->getDriverModel()->as<NEO::Wddm>()->getAdapterBDF();
|
|
|
|
if (bdfDevice1.Bus != bdfDevice2.Bus) {
|
|
return (bdfDevice1.Bus < bdfDevice2.Bus);
|
|
}
|
|
|
|
if (bdfDevice1.Device != bdfDevice2.Device) {
|
|
return (bdfDevice1.Device < bdfDevice2.Device);
|
|
}
|
|
|
|
return bdfDevice1.Function < bdfDevice2.Function;
|
|
}
|
|
|
|
void ExecutionEnvironment::sortNeoDevices() {
|
|
const auto pciOrderVar = DebugManager.flags.ZE_ENABLE_PCI_ID_DEVICE_ORDER.get();
|
|
if (pciOrderVar) {
|
|
std::sort(rootDeviceEnvironments.begin(), rootDeviceEnvironments.end(), comparePciIdBusNumber);
|
|
}
|
|
}
|
|
|
|
} // namespace NEO
|