mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-19 06:24:51 +08:00
c43233dabf
Cleaned up files: level_zero/core/source/kernel/kernel_hw.h shared/source/helpers/common_types.h shared/test/common/libult/linux/drm_mock.h shared/test/common/libult/ult_command_stream_receiver.h Related-To: NEO-5548 Signed-off-by: Warchulski, Jaroslaw <jaroslaw.warchulski@intel.com>
43 lines
1.3 KiB
C++
43 lines
1.3 KiB
C++
/*
|
|
* Copyright (C) 2019-2023 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "shared/source/helpers/debug_helpers.h"
|
|
#include "shared/source/helpers/device_bitfield.h"
|
|
#include "shared/source/helpers/non_copyable_or_moveable.h"
|
|
|
|
#include <atomic>
|
|
#include <memory>
|
|
|
|
namespace NEO {
|
|
class LocalMemoryUsageBankSelector : public NonCopyableOrMovableClass {
|
|
public:
|
|
LocalMemoryUsageBankSelector() = delete;
|
|
LocalMemoryUsageBankSelector(uint32_t banksCount);
|
|
uint32_t getLeastOccupiedBank(DeviceBitfield deviceBitfield);
|
|
void reserveOnBanks(uint32_t memoryBanks, uint64_t allocationSize) {
|
|
updateUsageInfo(memoryBanks, allocationSize, true);
|
|
}
|
|
void freeOnBanks(uint32_t memoryBanks, uint64_t allocationSize) {
|
|
updateUsageInfo(memoryBanks, allocationSize, false);
|
|
}
|
|
|
|
uint64_t getOccupiedMemorySizeForBank(uint32_t bankIndex) {
|
|
UNRECOVERABLE_IF(bankIndex >= banksCount);
|
|
return memorySizes[bankIndex].load();
|
|
}
|
|
|
|
protected:
|
|
uint32_t banksCount = 0;
|
|
std::unique_ptr<std::atomic<uint64_t>[]> memorySizes = nullptr;
|
|
void updateUsageInfo(uint32_t memoryBanks, uint64_t allocationSize, bool reserve);
|
|
void freeOnBank(uint32_t bankIndex, uint64_t allocationSize);
|
|
void reserveOnBank(uint32_t bankIndex, uint64_t allocationSize);
|
|
};
|
|
} // namespace NEO
|