Skip to content

Commit 70f2c2a

Browse files
authored
[Bugfix] Fix 'InductorAdaptor object has no attribute 'cache_dir' (#15674)
Signed-off-by: Jee Jee Li <pandaleefree@gmail.com>
1 parent 280d074 commit 70f2c2a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/compilation/compiler_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,7 @@ def compute_hash(self, vllm_config: VllmConfig) -> str:
144144
return hash_str
145145

146146
def initialize_cache(self, cache_dir: str, disable_cache: bool = False):
147+
self.cache_dir = cache_dir
147148
if disable_cache:
148149
return
149150
# redirect the cache directory to a sub-directory
@@ -156,7 +157,6 @@ def initialize_cache(self, cache_dir: str, disable_cache: bool = False):
156157
triton_cache = os.path.join(cache_dir, "triton_cache")
157158
os.makedirs(triton_cache, exist_ok=True)
158159
os.environ["TRITON_CACHE_DIR"] = triton_cache
159-
self.cache_dir = cache_dir
160160

161161
def compile(
162162
self,

0 commit comments

Comments
 (0)