From c3598c2823be38980569be9d96785209a4b7c919 Mon Sep 17 00:00:00 2001 From: Pawel Szczerbuk Date: Mon, 11 Mar 2024 08:57:31 -0700 Subject: [PATCH] Turn FileCacheManager.has_file back into an interface method --- python/triton/runtime/cache.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/triton/runtime/cache.py b/python/triton/runtime/cache.py index 65a000b038ba..2e8d70ea4ef8 100644 --- a/python/triton/runtime/cache.py +++ b/python/triton/runtime/cache.py @@ -68,20 +68,20 @@ def __init__(self, key, override=False, dump=False): def _make_path(self, filename) -> str: return os.path.join(self.cache_dir, filename) - def _has_file(self, filename) -> bool: + def has_file(self, filename) -> bool: if not self.cache_dir: raise RuntimeError("Could not create or locate cache dir") return os.path.exists(self._make_path(filename)) def get_file(self, filename) -> Optional[str]: - if self._has_file(filename): + if self.has_file(filename): return self._make_path(filename) else: return None def get_group(self, filename: str) -> Optional[Dict[str, str]]: grp_filename = f"__grp__{filename}" - if not self._has_file(grp_filename): + if not self.has_file(grp_filename): return None grp_filepath = self._make_path(grp_filename) with open(grp_filepath) as f: