Skip to content

Commit d539dfa

Browse files
committed
[ET-VK] Allow clients to specify cache_data_path
Pull Request resolved: #11350 TSIA ghstack-source-id: 288078561 @exported-using-ghexport Differential Revision: [D75918105](https://our.internmc.facebook.com/intern/diff/D75918105/)
1 parent 64ef38e commit d539dfa

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

backends/vulkan/runtime/vk_api/Runtime.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -258,7 +258,8 @@ uint32_t select_first(const std::vector<Runtime::DeviceMapping>& devices) {
258258
// Global runtime initialization
259259
//
260260

261-
std::unique_ptr<Runtime> init_global_vulkan_runtime() {
261+
std::unique_ptr<Runtime> init_global_vulkan_runtime(
262+
const std::string& cache_data_path) {
262263
// Load Vulkan drivers
263264
#if defined(USE_VULKAN_VOLK)
264265
if (VK_SUCCESS != volkInitialize()) {
@@ -278,7 +279,6 @@ std::unique_ptr<Runtime> init_global_vulkan_runtime() {
278279
#endif /* VULKAN_DEBUG */
279280
const bool init_default_device = true;
280281
const uint32_t num_requested_queues = 1; // TODO: raise this value
281-
const std::string cache_data_path = ""; // TODO: expose to client
282282

283283
const RuntimeConfig default_config{
284284
enable_validation_messages,
@@ -377,13 +377,13 @@ uint32_t Runtime::create_adapter(const Selector& selector) {
377377
return adapter_i;
378378
}
379379

380-
Runtime* runtime() {
380+
Runtime* runtime(const std::string& cache_data_path) {
381381
// The global vulkan runtime is declared as a static local variable within a
382382
// non-static function to ensure it has external linkage. If it were a global
383383
// static variable there would be one copy per translation unit that includes
384384
// Runtime.h as it would have internal linkage.
385385
static const std::unique_ptr<Runtime> p_runtime =
386-
init_global_vulkan_runtime();
386+
init_global_vulkan_runtime(cache_data_path);
387387

388388
VK_CHECK_COND(
389389
p_runtime,

backends/vulkan/runtime/vk_api/Runtime.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ class Runtime final {
102102

103103
// The global runtime is retrieved using this function, where it is declared as
104104
// a static local variable.
105-
Runtime* runtime();
105+
Runtime* runtime(const std::string& cache_data_path = "");
106106

107107
} // namespace vkapi
108108
} // namespace vkcompute

0 commit comments

Comments
 (0)