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
|
/*
* Copyright (C) 2019-2025 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "shared/source/helpers/non_copyable_or_moveable.h"
#include "shared/source/os_interface/os_handle.h"
#include "shared/source/utilities/arrayref.h"
#include <cstdint>
#include <memory>
#include <mutex>
#include <string>
#include <unordered_map>
namespace NEO {
struct HardwareInfo;
struct CompilerCacheConfig {
bool enabled = false;
std::string cacheFileExtension;
std::string cacheDir;
size_t cacheSize = 0;
};
class CompilerCache : NEO::NonCopyableAndNonMovableClass {
public:
CompilerCache(const CompilerCacheConfig &config);
virtual ~CompilerCache() = default;
const CompilerCacheConfig &getConfig() {
return config;
}
const std::string getCachedFileName(const HardwareInfo &hwInfo, ArrayRef<const char> input,
ArrayRef<const char> options, ArrayRef<const char> internalOptions,
ArrayRef<const char> specIds, ArrayRef<const char> specValues,
ArrayRef<const char> igcRevision, size_t igcLibSize, time_t igcLibMTime);
MOCKABLE_VIRTUAL bool cacheBinary(const std::string &kernelFileHash, const char *pBinary, size_t binarySize);
MOCKABLE_VIRTUAL std::unique_ptr<char[]> loadCachedBinary(const std::string &kernelFileHash, size_t &cachedBinarySize);
protected:
MOCKABLE_VIRTUAL bool evictCache(uint64_t &bytesEvicted);
MOCKABLE_VIRTUAL bool renameTempFileBinaryToProperName(const std::string &oldName, const std::string &kernelFileHash);
MOCKABLE_VIRTUAL bool createUniqueTempFileAndWriteData(char *tmpFilePathTemplate, const char *pBinary, size_t binarySize);
MOCKABLE_VIRTUAL void lockConfigFileAndReadSize(const std::string &configFilePath, UnifiedHandle &fd, size_t &directorySize);
static std::mutex cacheAccessMtx;
CompilerCacheConfig config;
};
static_assert(NEO::NonCopyableAndNonMovable<CompilerCache>);
} // namespace NEO
|
