diff --git a/python/triton/runtime/cache.py b/python/triton/runtime/cache.py index 82b2fea37e9b..62895508b019 100644 --- a/python/triton/runtime/cache.py +++ b/python/triton/runtime/cache.py @@ -256,9 +256,9 @@ def put_group(self, filename: str, group: Dict[str, str]): __cache_cls_nme = "DEFAULT" -def _base64(key): +def _base32(key): # Assume key is a hex string. - return base64.urlsafe_b64encode(bytes.fromhex(key)).decode("utf-8").rstrip("=") + return base64.b32encode(bytes.fromhex(key)).decode("utf-8").rstrip("=") def get_cache_manager(key) -> CacheManager: @@ -274,15 +274,15 @@ def get_cache_manager(key) -> CacheManager: __cache_cls = getattr(module, clz_nme) __cache_cls_nme = user_cache_manager - return __cache_cls(_base64(key)) + return __cache_cls(_base32(key)) def get_override_manager(key) -> CacheManager: - return __cache_cls(_base64(key), override=True) + return __cache_cls(_base32(key), override=True) def get_dump_manager(key) -> CacheManager: - return __cache_cls(_base64(key), dump=True) + return __cache_cls(_base32(key), dump=True) def make_so_cache_key(version_hash, signature, constants, ids, **kwargs): @@ -292,4 +292,4 @@ def make_so_cache_key(version_hash, signature, constants, ids, **kwargs): for kw in kwargs: key = f"{key}-{kwargs.get(kw)}" key = hashlib.sha256(key.encode("utf-8")).hexdigest() - return _base64(key) + return _base32(key)