Unverified Commit 70f2c2a7 authored by Jee Jee Li's avatar Jee Jee Li Committed by GitHub
Browse files

[Bugfix] Fix 'InductorAdaptor object has no attribute 'cache_dir' (#15674)


Signed-off-by: default avatarJee Jee Li <pandaleefree@gmail.com>
parent 280d0741
......@@ -144,6 +144,7 @@ class InductorAdaptor(CompilerInterface):
return hash_str
def initialize_cache(self, cache_dir: str, disable_cache: bool = False):
self.cache_dir = cache_dir
if disable_cache:
return
# redirect the cache directory to a sub-directory
......@@ -156,7 +157,6 @@ class InductorAdaptor(CompilerInterface):
triton_cache = os.path.join(cache_dir, "triton_cache")
os.makedirs(triton_cache, exist_ok=True)
os.environ["TRITON_CACHE_DIR"] = triton_cache
self.cache_dir = cache_dir
def compile(
self,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment