We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 26a121b commit 5cf02fdCopy full SHA for 5cf02fd
lightllm/common/triton_utils/autotuner.py
@@ -63,6 +63,7 @@ def __init__(
63
get_current_device_name(),
64
self.kernel_name,
65
)
66
+ os.makedirs(self.cache_dir, exist_ok=True)
67
self.fn = fn
68
self.static_key_func = static_key_func
69
self.run_key_func = run_key_func
@@ -83,7 +84,6 @@ def _try_load_cache(self, static_key):
83
84
if static_key in self.cached_configs:
85
return
86
- os.makedirs(self.cache_dir, exist_ok=True)
87
cache_file = os.path.join(self.cache_dir, KernelConfigs.get_config_file_name(static_key))
88
if os.path.exists(cache_file):
89
with open(cache_file, "rb") as f:
0 commit comments