C++ jit

Signed-off-by: Mayank Mishra <[email protected]>
mayank31398 · Dec 21, 2024 · 4fe469d · 4fe469d
1 parent 74c242d
commit 4fe469d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cute_kernels/cutotune/cache.py b/cute_kernels/cutotune/cache.py
@@ -51,6 +51,9 @@ def load(self) -> None:
         self.best_cache = self._deserialize(cache["best_configs"], False)
 
     def get_best_configs(self, function_hash: str) -> dict[str, CutoTuneConfig]:
+        if function_hash not in self.best_cache:
+            self.best_cache[function_hash] = {}
+
         return self.best_cache[function_hash]
 
     def _serialize(self, x: dict, has_config_time_list: bool) -> dict: