xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-19 06:05:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Signed-off-by: Jee Jee Li <pandaleefree@gmail.com>
This commit is contained in:
Jee Jee Li 2025-03-28 17:10:40 +08:00 committed by GitHub
parent 280d074103
commit 70f2c2a709
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/compilation/compiler_interface.py
View File

@ -144,6 +144,7 @@ class InductorAdaptor(CompilerInterface):
return hash_str return hash_str
def initialize_cache(self, cache_dir: str, disable_cache: bool = False): def initialize_cache(self, cache_dir: str, disable_cache: bool = False):
self.cache_dir = cache_dir
if disable_cache: if disable_cache:
return return
# redirect the cache directory to a sub-directory # redirect the cache directory to a sub-directory
@ -156,7 +157,6 @@ class InductorAdaptor(CompilerInterface):
triton_cache = os.path.join(cache_dir, "triton_cache") triton_cache = os.path.join(cache_dir, "triton_cache")
os.makedirs(triton_cache, exist_ok=True) os.makedirs(triton_cache, exist_ok=True)
os.environ["TRITON_CACHE_DIR"] = triton_cache os.environ["TRITON_CACHE_DIR"] = triton_cache
self.cache_dir = cache_dir
def compile( def compile(
self, self,