biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[BugFix] Hot fix in setup.py for neuron build (#3537)

Browse Source
This commit is contained in:
Zhuohan Li
2024-03-20 17:59:52 -07:00
committed by GitHub
parent f1c0fc3919
commit 523e30ea0c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -168,7 +168,7 @@ class cmake_build_ext(build_ext):
def _is_cuda() -> bool:
return torch.version.cuda is not None
return torch.version.cuda is not None and not _is_neuron()
def _is_hip() -> bool: