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

Update setup.py (#282)

Browse Source 
Co-authored-by: neubig <neubig@gmail.com>
This commit is contained in:
Cody Yu 2023-06-27 14:34:23 -07:00 committed by GitHub
parent 4026a049d3
commit 2b7d3aca2e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
setup.py
View File

@ -20,10 +20,9 @@ ABI = 1 if torch._C._GLIBCXX_USE_CXX11_ABI else 0
CXX_FLAGS += [f"-D_GLIBCXX_USE_CXX11_ABI={ABI}"] CXX_FLAGS += [f"-D_GLIBCXX_USE_CXX11_ABI={ABI}"]
NVCC_FLAGS += [f"-D_GLIBCXX_USE_CXX11_ABI={ABI}"] NVCC_FLAGS += [f"-D_GLIBCXX_USE_CXX11_ABI={ABI}"]
if not torch.cuda.is_available(): if CUDA_HOME is None:
raise RuntimeError( raise RuntimeError(
f"Cannot find CUDA at CUDA_HOME: {CUDA_HOME}. " f"Cannot find CUDA_HOME. CUDA must be available in order to build the package.")
"CUDA must be available in order to build the package.")
def get_nvcc_cuda_version(cuda_dir: str) -> Version: def get_nvcc_cuda_version(cuda_dir: str) -> Version: