20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 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
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
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: