Skip to content

GitLab

Unverified Commit 11f44927 authored by Jinze (Richard) Xue's avatar Jinze (Richard) Xue Committed by GitHub
Browse files

Pin cub to 1.11.0 for cuda 10.2 (#601) 

* Fix cub to 1.11.0 for cuda 10.2

* quick save

* quick save
parent d1a63639
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
setup.py
View file @ 11f44927
...@@ -39,12 +39,12 @@ def maybe_download_cub(): ...@@ -39,12 +39,12 @@ def maybe_download_cub():
os.makedirs('include') os.makedirs('include')
commands = """ commands = """
echo "Downloading CUB library"; echo "Downloading CUB library";
wget -q https://github.com/NVIDIA/cub/archive/main.zip; wget -q https://github.com/NVIDIA/cub/archive/refs/tags/1.11.0.zip;
unzip -q main.zip -d include; unzip -q 1.11.0.zip -d include;
mv include/cub-main/cub include; mv include/cub-1.11.0/cub include;
echo "Removing unnecessary files"; echo "Removing unnecessary files";
rm main.zip; rm 1.11.0.zip;
rm -rf include/cub-main; rm -rf include/cub-1.11.0;
""" """
subprocess.run(commands, shell=True, check=True, universal_newlines=True) subprocess.run(commands, shell=True, check=True, universal_newlines=True)
return [os.path.abspath("./include")] return [os.path.abspath("./include")]
......
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