Skip to content

GitLab

"...composable_kernel_onnxruntime.git" did not exist on "1a0cd5d160dfbe107a454f975a26599fc6daddd4"
Unverified Commit 2b7d3aca authored by Cody Yu's avatar Cody Yu Committed by GitHub
Browse files

Update setup.py (#282) 


Co-authored-by: default avatarneubig <neubig@gmail.com>
parent 4026a049
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
setup.py
View file @ 2b7d3aca
...@@ -20,10 +20,9 @@ ABI = 1 if torch._C._GLIBCXX_USE_CXX11_ABI else 0 ...@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment