From e58919b13dcc3ef967e7e3d4f061d6c6a38cd176 Mon Sep 17 00:00:00 2001 From: Joe Runde Date: Mon, 3 Jun 2024 10:20:43 -0600 Subject: [PATCH] Update triton_patch/custom_cache_manager.py Co-authored-by: Nick Hill Signed-off-by: Joe Runde --- triton_patch/custom_cache_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/triton_patch/custom_cache_manager.py b/triton_patch/custom_cache_manager.py index a923d9e63..902c0ee1f 100644 --- a/triton_patch/custom_cache_manager.py +++ b/triton_patch/custom_cache_manager.py @@ -21,7 +21,7 @@ def __init__(self, key, override=False, dump=False): # create cache directory if it doesn't exist self.cache_dir = os.getenv("TRITON_CACHE_DIR", "").strip() or default_cache_dir() - self.cache_dir = self.cache_dir + "_" + str(os.getpid()) + self.cache_dir = f"{self.cache_dir}_{os.getpid()}" print(f"{self.cache_dir=}") if self.cache_dir: self.cache_dir = os.path.join(self.cache_dir, self.key)