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
|
/*
* Copyright (C) 2021-2025 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include <cstddef>
#include <cstdint>
#include <vector>
namespace NEO {
enum class HeapIndex : uint32_t;
class AlignmentSelector {
public:
constexpr static inline float anyWastage = 1.f;
constexpr static inline float noWastage = 0.f;
struct CandidateAlignment {
size_t alignment; // selected alignment for the allocation
bool applyForSmallerSize; // if enabled, this alignment can be used even when size < alignment
float maxMemoryWastage; // maximum percent of wasted memory after alignUp(size, alignment)
HeapIndex heap; // GPUVA heap, which allocation should go to
};
AlignmentSelector() = default;
void addCandidateAlignment(size_t alignment, bool applyForSmallerSize, float maxMemoryWastage);
void addCandidateAlignment(size_t alignment, bool applyForSmallerSize, float maxMemoryWastage, HeapIndex heap);
CandidateAlignment selectAlignment(size_t size) const;
CandidateAlignment selectAlignment(size_t size, size_t maximumPossibleAlignment) const;
auto &peekCandidateAlignments() const { return candidateAlignments; }
private:
std::vector<CandidateAlignment> candidateAlignments; // sorted by alignment size
};
bool operator==(const AlignmentSelector::CandidateAlignment &left, const AlignmentSelector::CandidateAlignment &right);
} // namespace NEO
|
