1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
/*
* Copyright (C) 2021-2024 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "shared/source/helpers/device_bitfield.h"
#include <bitset>
#include <limits>
#include <vector>
namespace NEO {
class AffinityMaskHelper {
public:
using AffinityMaskContainer = std::vector<std::bitset<4>>;
AffinityMaskHelper(bool allSubdevicesActive) {
if (!allSubdevicesActive) {
return;
}
constexpr size_t maxInitialSubdeviceCount = 4;
enableAllGenericSubDevices(maxInitialSubdeviceCount);
}
AffinityMaskHelper() : AffinityMaskHelper(false) {}
void enableGenericSubDevice(uint32_t subDeviceIndex) {
enableGenericSubDevice(subDeviceIndex, std::numeric_limits<uint32_t>::max());
}
void enableAllGenericSubDevices(uint32_t subDeviceCount) {
for (uint32_t i = 0; i < subDeviceCount; i++) {
enableGenericSubDevice(i);
}
}
DeviceBitfield getGenericSubDevicesMask() const {
return genericSubDevicesMask;
}
DeviceBitfield getEnginesMask(uint32_t subDeviceIndex) const {
return subDevicesWithEnginesMasks[subDeviceIndex];
}
bool isDeviceEnabled() const {
return genericSubDevicesMask.any();
}
protected:
void enableGenericSubDevice(uint32_t subDeviceIndex, uint32_t enginesMask) {
if ((subDeviceIndex + 1) > subDevicesWithEnginesMasks.size()) {
subDevicesWithEnginesMasks.resize(subDeviceIndex + 1);
subDevicesWithEnginesMasks[subDeviceIndex] = 0;
}
genericSubDevicesMask.set(subDeviceIndex);
subDevicesWithEnginesMasks[subDeviceIndex] |= enginesMask;
}
AffinityMaskContainer subDevicesWithEnginesMasks;
DeviceBitfield genericSubDevicesMask = 0;
};
} // namespace NEO
|
