Skip to content

GitLab

Unverified Commit 9fe47355 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

[Build] Option to disable CUDA arch autodetection for PyTorch (#2454)

parent 9a7235fa
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 4 deletions
tensoradapter/pytorch/build.bat
View file @ 9fe47355
...@@ -11,7 +11,7 @@ IF x%1x == xx GOTO single ...@@ -11,7 +11,7 @@ IF x%1x == xx GOTO single
FOR %%X IN (%*) DO ( FOR %%X IN (%*) DO (
DEL /S /Q * DEL /S /Q *
"%CMAKE_COMMAND%" -DCMAKE_CONFIGURATION_TYPES=Release -DCUDA_TOOLKIT_ROOT_DIR="%CUDA_TOOLKIT_ROOT_DIR%" -DPYTHON_INTERP=%%X .. -G "Visual Studio 16 2019" || EXIT /B 1 "%CMAKE_COMMAND%" -DCMAKE_CONFIGURATION_TYPES=Release -DCUDA_TOOLKIT_ROOT_DIR="%CUDA_TOOLKIT_ROOT_DIR%" -DTORCH_CUDA_ARCH_LIST=%TORCH_CUDA_ARCH_LIST% -DPYTHON_INTERP=%%X .. -G "Visual Studio 16 2019" || EXIT /B 1
msbuild tensoradapter_pytorch.sln /m /nr:false || EXIT /B 1 msbuild tensoradapter_pytorch.sln /m /nr:false || EXIT /B 1
COPY /Y Release\*.dll "%BINDIR%\tensoradapter\pytorch" || EXIT /B 1 COPY /Y Release\*.dll "%BINDIR%\tensoradapter\pytorch" || EXIT /B 1
) )
...@@ -21,7 +21,7 @@ GOTO end ...@@ -21,7 +21,7 @@ GOTO end
:single :single
DEL /S /Q * DEL /S /Q *
"%CMAKE_COMMAND%" -DCMAKE_CONFIGURATION_TYPES=Release -DCUDA_TOOLKIT_ROOT_DIR="%CUDA_TOOLKIT_ROOT_DIR%" .. -G "Visual Studio 16 2019" || EXIT /B 1 "%CMAKE_COMMAND%" -DCMAKE_CONFIGURATION_TYPES=Release -DCUDA_TOOLKIT_ROOT_DIR="%CUDA_TOOLKIT_ROOT_DIR%" -DTORCH_CUDA_ARCH_LIST=%TORCH_CUDA_ARCH_LIST% .. -G "Visual Studio 16 2019" || EXIT /B 1
msbuild tensoradapter_pytorch.sln /m /nr:false || EXIT /B 1 msbuild tensoradapter_pytorch.sln /m /nr:false || EXIT /B 1
COPY /Y Release\*.dll "%BINDIR%\tensoradapter\pytorch" || EXIT /B 1 COPY /Y Release\*.dll "%BINDIR%\tensoradapter\pytorch" || EXIT /B 1
......
tensoradapter/pytorch/build.sh
View file @ 9fe47355
...@@ -7,14 +7,16 @@ mkdir -p build ...@@ -7,14 +7,16 @@ mkdir -p build
mkdir -p $BINDIR/tensoradapter/pytorch mkdir -p $BINDIR/tensoradapter/pytorch
cd build cd build
CMAKE_FLAGS="-DCUDA_TOOLKIT_ROOT_DIR=$CUDA_TOOLKIT_ROOT_DIR -DTORCH_CUDA_ARCH_LIST=$TORCH_CUDA_ARCH_LIST"
if [ $# -eq 0 ]; then if [ $# -eq 0 ]; then
${CMAKE_COMMAND} -DCUDA_TOOLKIT_ROOT_DIR=$CUDA_TOOLKIT_ROOT_DIR .. $CMAKE_COMMAND $CMAKE_FLAGS ..
make -j make -j
cp -v *.so $BINDIR/tensoradapter/pytorch cp -v *.so $BINDIR/tensoradapter/pytorch
else else
for PYTHON_INTERP in $@; do for PYTHON_INTERP in $@; do
rm -rf * rm -rf *
${CMAKE_COMMAND} -DCUDA_TOOLKIT_ROOT_DIR=$CUDA_TOOLKIT_ROOT_DIR -DPYTHON_INTERP=$PYTHON_INTERP .. $CMAKE_COMMAND $CMAKE_FLAGS -DPYTHON_INTERP=$PYTHON_INTERP ..
make -j make -j
cp -v *.so $BINDIR/tensoradapter/pytorch cp -v *.so $BINDIR/tensoradapter/pytorch
done done
......
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