50 lines
1.6 KiB
C++
50 lines
1.6 KiB
C++
/*
|
|
* Copyright (C) 2022-2023 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/helpers/vec.h"
|
|
#include "shared/source/utilities/stackvec.h"
|
|
|
|
#include <array>
|
|
#include <mutex>
|
|
|
|
namespace NEO {
|
|
class GfxCoreHelper;
|
|
class LocalIdsCache {
|
|
public:
|
|
struct LocalIdsCacheEntry {
|
|
Vec3<uint16_t> groupSize = {0, 0, 0};
|
|
uint8_t *localIdsData = nullptr;
|
|
size_t localIdsSize = 0U;
|
|
size_t localIdsSizeAllocated = 0U;
|
|
size_t accessCounter = 0;
|
|
};
|
|
|
|
LocalIdsCache() = delete;
|
|
LocalIdsCache(LocalIdsCache &) = delete;
|
|
LocalIdsCache &operator=(const LocalIdsCache &other) = delete;
|
|
|
|
LocalIdsCache(size_t cacheSize, std::array<uint8_t, 3> wgDimOrder, uint8_t simdSize, uint8_t grfSize, bool usesOnlyImages = false);
|
|
~LocalIdsCache();
|
|
|
|
void setLocalIdsForGroup(const Vec3<uint16_t> &group, void *destination, const GfxCoreHelper &gfxCoreHelper);
|
|
size_t getLocalIdsSizeForGroup(const Vec3<uint16_t> &group, const GfxCoreHelper &gfxCoreHelper) const;
|
|
size_t getLocalIdsSizePerThread() const;
|
|
|
|
protected:
|
|
void setLocalIdsForEntry(LocalIdsCacheEntry &entry, void *destination);
|
|
void commitNewEntry(LocalIdsCacheEntry &entry, const Vec3<uint16_t> &group, const GfxCoreHelper &gfxCoreHelper);
|
|
std::unique_lock<std::mutex> lock();
|
|
|
|
StackVec<LocalIdsCacheEntry, 4> cache;
|
|
std::mutex setLocalIdsMutex;
|
|
const std::array<uint8_t, 3> wgDimOrder;
|
|
const uint32_t localIdsSizePerThread;
|
|
const uint8_t grfSize;
|
|
const uint8_t simdSize;
|
|
const bool usesOnlyImages;
|
|
};
|
|
} // namespace NEO
|