feat: add kernel preload

2025-12-27 11:07:52 +00:00
parent ffe2b6b974
commit d763aaf304
3 changed files with 61 additions and 0 deletions
@@ -42,6 +42,13 @@ public:
        Compiler::library_version = get_library_version();
    }

+    static std::filesystem::path get_cache_dir() {
+        std::filesystem::path cache_dir = std::filesystem::path(get_env<std::string>("HOME")) / ".deep_gemm";
+        if (const auto& env_cache_dir = get_env<std::string>("DG_JIT_CACHE_DIR"); !env_cache_dir.empty())
+            cache_dir = env_cache_dir;
+        return cache_dir;
+    }
+
    std::string signature, flags;
    std::filesystem::path cache_dir_path;