add CUDA_ARCH="8.0;12.0"

WindowsWhlBuilder.bat

@@ -1,56 +1,79 @@
-@ECHO off
-
-if [%1] == [test] shift & goto :test
-if [%1] == [build] shift & goto :build
-if [%1] == [clean] shift & goto :clean
-
-:parseArgs
-if [%1] == [WORKERS] set NATTEN_N_WORKERS=%2 & shift & shift & goto :parseargs
-if [%1] == [CUDA_ARCH] set NATTEN_CUDA_ARCH=%2 & shift & shift & goto :parseargs
-if [%1] == [FORCE_CUDA] set NATTEN_WITH_CUDA=%2 & shift & shift & goto :parseargs
-if [%1] == [VERBOSE] set NATTEN_VERBOSE=%2 & shift & shift & goto :parseargs
-goto :buildContinue
-:end
-
-:test
-echo "Testing NATTEN"
-pip install -r requirements-dev.txt
-pytest -v -x ./tests
-goto :eof
-:end
-
-:build
-goto :buildStart
-:end
-
-:buildStart
-goto :parseargs
-:end
-
-:buildFinalize
-set NATTEN_N_WORKERS=
-set NATTEN_CUDA_ARCH=
-set NATTEN_VERBOSE=
-set NATTEN_IS_BUILDING_DIST=
-set NATTEN_WITH_CUDA=
-goto :eof
-:end
-
-:buildContinue
-echo NATTEN_N_WORKERS: %NATTEN_N_WORKERS%
-echo NATTEN_CUDA_ARCH: %NATTEN_CUDA_ARCH%
-echo NATTEN_WITH_CUDA: %NATTEN_WITH_CUDA%
-echo NATTEN_VERBOSE: %NATTEN_VERBOSE%
-set NATTEN_IS_BUILDING_DIST=1
-rem set CUDA_ARCH=8.6
-pip install -r requirements.txt
-python setup.py bdist_wheel
-goto :buildFinalize
-:end
-
-:clean
-echo Cleaning up
-echo "Removing %CD%\build"
-del %CD%\build
-goto :eof
-:end
+@ECHO off
+
+setlocal
+
+if [%1] == [test] shift & goto :test
+if [%1] == [build] shift & goto :build
+if [%1] == [clean] shift & goto :clean
+
+:parseArgs
+if "%~1" == "WORKERS" (
+    set "NATTEN_N_WORKERS=%~2"
+    shift & shift
+    goto :parseArgs
+)
+if "%~1" == "CUDA_ARCH" (
+    set "NATTEN_CUDA_ARCH=%~2"
+    shift & shift
+    goto :parseArgs
+)
+if "%~1" == "FORCE_CUDA" (
+    set "NATTEN_WITH_CUDA=%~2"
+    shift & shift
+    goto :parseArgs
+)
+if "%~1" == "VERBOSE" (
+    set "NATTEN_VERBOSE=%~2"
+    shift & shift
+    goto :parseArgs
+)
+goto :buildContinue
+:end
+
+:test
+echo "Testing NATTEN"
+pip install -r requirements-dev.txt
+pytest -v -x ./tests
+goto :eof
+:end
+
+:build
+goto :buildStart
+:end
+
+:buildStart
+goto :parseargs
+:end
+
+:buildFinalize
+set NATTEN_N_WORKERS=
+set NATTEN_CUDA_ARCH=
+set NATTEN_VERBOSE=
+set NATTEN_IS_BUILDING_DIST=
+set NATTEN_WITH_CUDA=
+goto :eof
+:end
+
+:buildContinue
+echo NATTEN_N_WORKERS: %NATTEN_N_WORKERS%
+echo NATTEN_CUDA_ARCH: %NATTEN_CUDA_ARCH%
+echo NATTEN_WITH_CUDA: %NATTEN_WITH_CUDA%
+echo NATTEN_VERBOSE: %NATTEN_VERBOSE%
+set NATTEN_IS_BUILDING_DIST=1
+rem set NATTEN_CUDA_ARCH=8.0;12.0
+pip install -r requirements.txt
+python setup.py bdist_wheel
+goto :buildFinalize
+:end
+
+:clean
+echo Cleaning up
+if exist "%CD%\build" (
+    echo "Removing %CD%\build"
+    rmdir /s /q "%CD%\build"
+) else (
+    echo "%CD%\build does not exist, ignored"
+)
+goto :eof
+:end
+endlocal