Skip to content

Commit

Permalink
Update triton_patch/custom_cache_manager.py
Browse files Browse the repository at this point in the history
Co-authored-by: Nick Hill <nickhill@us.ibm.com>
Signed-off-by: Joe Runde <joe@joerun.de>
  • Loading branch information
joerunde and njhill authored Jun 3, 2024
1 parent cf4eae9 commit e58919b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion triton_patch/custom_cache_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit e58919b

Please sign in to comment.