wehub-resource-sync
32 lines
1006 B
Batchfile
32 lines
1006 B
Batchfile
REM Helper script to build tensor adapter libraries for PyTorch
|
|
@ECHO OFF
|
|
SETLOCAL EnableDelayedExpansion
|
|
|
|
MD "%BINDIR%\tensoradapter\pytorch"
|
|
DEL /S /Q build
|
|
MD build
|
|
PUSHD build
|
|
|
|
IF x%1x == xx GOTO single
|
|
|
|
FOR %%X IN (%*) DO (
|
|
DEL /S /Q *
|
|
"%CMAKE_COMMAND%" -DCMAKE_CONFIGURATION_TYPES=Release -DCUDA_TOOLKIT_ROOT_DIR="%CUDA_TOOLKIT_ROOT_DIR%" -DTORCH_CUDA_ARCH_LIST=%TORCH_CUDA_ARCH_LIST% -DUSE_CUDA=%USE_CUDA% -DPYTHON_INTERP=%%X .. -G "Visual Studio 16 2019" || EXIT /B 1
|
|
msbuild tensoradapter_pytorch.sln /m /nr:false || EXIT /B 1
|
|
COPY /Y Release\*.dll "%BINDIR%\tensoradapter\pytorch" || EXIT /B 1
|
|
)
|
|
|
|
GOTO end
|
|
|
|
:single
|
|
|
|
DEL /S /Q *
|
|
"%CMAKE_COMMAND%" -DCMAKE_CONFIGURATION_TYPES=Release -DCUDA_TOOLKIT_ROOT_DIR="%CUDA_TOOLKIT_ROOT_DIR%" -DTORCH_CUDA_ARCH_LIST=%TORCH_CUDA_ARCH_LIST% -DUSE_CUDA=%USE_CUDA% .. -G "Visual Studio 16 2019" || EXIT /B 1
|
|
msbuild tensoradapter_pytorch.sln /m /nr:false || EXIT /B 1
|
|
COPY /Y Release\*.dll "%BINDIR%\tensoradapter\pytorch" || EXIT /B 1
|
|
|
|
:end
|
|
POPD
|
|
|
|
ENDLOCAL
|