Skip to content

Commit d305652

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent b1fc106 commit d305652

File tree

9 files changed

+9
-11
lines changed

9 files changed

+9
-11
lines changed

.github/workflows/cuda/cu116-Linux-env.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -5,4 +5,4 @@ LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
55
PATH=${CUDA_HOME}/bin:${PATH}
66

77
export FORCE_CUDA=1
8-
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
8+
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"

.github/workflows/cuda/cu116-Linux.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -12,4 +12,4 @@ sudo apt-get -qq update
1212
sudo apt install cuda-nvcc-11-6 cuda-libraries-dev-11-6
1313
sudo apt clean
1414

15-
rm -f https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda-repo-${OS}-11-6-local_11.6.2-510.47.03-1_amd64.deb
15+
rm -f https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda-repo-${OS}-11-6-local_11.6.2-510.47.03-1_amd64.deb

.github/workflows/cuda/cu116-Windows-env.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -5,4 +5,4 @@ PATH=${CUDA_HOME}/bin:$PATH
55
PATH=/c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/MSBuild/15.0/Bin:$PATH
66

77
export FORCE_CUDA=1
8-
export TORCH_CUDA_ARCH_LIST="6.0+PTX"
8+
export TORCH_CUDA_ARCH_LIST="6.0+PTX"

.github/workflows/cuda/cu116-Windows.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -17,4 +17,4 @@ echo ""
1717
echo "Installing from ${CUDA_FILE}..."
1818
PowerShell -Command "Start-Process -FilePath \"${CUDA_FILE}\" -ArgumentList \"-s nvcc_${CUDA_SHORT} cuobjdump_${CUDA_SHORT} nvprune_${CUDA_SHORT} cupti_${CUDA_SHORT} cublas_dev_${CUDA_SHORT} cudart_${CUDA_SHORT} cufft_dev_${CUDA_SHORT} curand_dev_${CUDA_SHORT} cusolver_dev_${CUDA_SHORT} cusparse_dev_${CUDA_SHORT} thrust_${CUDA_SHORT} npp_dev_${CUDA_SHORT} nvrtc_dev_${CUDA_SHORT} nvml_dev_${CUDA_SHORT}\" -Wait -NoNewWindow"
1919
echo "Done!"
20-
rm -f "${CUDA_FILE}"
20+
rm -f "${CUDA_FILE}"

conda/torch-points-kernels/README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@
22
./build_conda.sh 3.9 1.9.0 cu111 # python, pytorch and cuda version
33
```
44

5-
These conda scripts are based off of [pytorch_sparse](https://github.com/rusty1s/pytorch_sparse/tree/master/conda/pytorch-sparse), track changes there to update.
5+
These conda scripts are based off of [pytorch_sparse](https://github.com/rusty1s/pytorch_sparse/tree/master/conda/pytorch-sparse), track changes there to update.

conda/torch-points-kernels/bld.bat

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,4 @@ copy "C:\\Program Files (x86)\\Microsoft Visual Studio\\2019\\Enterprise\\VC\\To
44
if errorlevel 1 exit 1
55

66
"%PYTHON%" -m pip install .
7-
if errorlevel 1 exit 1
7+
if errorlevel 1 exit 1
+1-1
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
11
"%PYTHON%" -m unittest
2-
if errorlevel 1 exit 1
2+
if errorlevel 1 exit 1
+1-1
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
$PYTHON -m unittest
1+
$PYTHON -m unittest

setup.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,7 @@ def get_ext_modules():
3333
extra_compile_args["cxx"] += ["-DVERSION_GE_1_3"]
3434

3535
ext_src_root = "cuda"
36-
ext_sources = glob.glob("{}/src/*.cpp".format(ext_src_root)) + glob.glob(
37-
"{}/src/*.cu".format(ext_src_root)
38-
)
36+
ext_sources = glob.glob("{}/src/*.cpp".format(ext_src_root)) + glob.glob("{}/src/*.cu".format(ext_src_root))
3937

4038
ext_modules = []
4139
if WITH_CUDA:

0 commit comments

Comments
 (0)