Small updates from review

Co-authored-by: Nick Hill <[email protected]>
Signed-off-by: Nick Hill <[email protected]>
Signed-off-by: Joe Runde <[email protected]>

Author: joerunde

triton_patch/custom_cache_manager.py

@@ -21,11 +21,12 @@ 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())
-            print(f"{self.cache_dir=}")
             if self.cache_dir:
+                self.cache_dir = f"{self.cache_dir}_{os.getpid()}"
                 self.cache_dir = os.path.join(self.cache_dir, self.key)
                 self.lock_path = os.path.join(self.cache_dir, "lock")
                 os.makedirs(self.cache_dir, exist_ok=True)
             else:
                 raise RuntimeError("Could not create or locate cache dir")
+
+        print(f"Triton cache dir: {self.cache_dir=}")