Add LibTorch support (peterjc123#50)

* Add libtorch support

* Remove empty line

* Correct paths and changed readme

* Fix cuda url

Author: peterjc123

.github/azure-pipelines.yml

@@ -11,6 +11,13 @@ jobs:
   timeoutInMinutes: 0
   cancelTimeoutInMinutes: 5
   condition: eq(variables['build.pytorch.enabled'], 'true')
+  strategy:
+    maxParallel: 10
+    matrix:
+      LIBTORCH:
+        BUILD_PYTHONLESS: 1
+      MAIN:
+        BUILD_PYTHONLESS: ''
 
   pool:
     vmImage: 'vs2017-win2016'

README.md

@@ -53,6 +53,13 @@ An additional CI is setup to run every night so that you can also get the latest
 | Windows x64 CUDA 9.0 (Conda)   | [![Build status](https://dev.azure.com/pytorch/PyTorch/_apis/build/status/peterjc123.builder?branchName=master&jobName=Windows_CUDA90_Conda_Build&configuration=PY3.5)](https://dev.azure.com/peterjc123/46ef0f8e-f34b-4a52-b6e0-c1000caf14a1/_apis/git/repositories/49b30ad3-8e9b-43ff-bbc8-87179a36d246/Items?path=%2Fconda%2Fpytorch-nightly-1.0-py3.5_cuda90_cudnn7he774522_1.tar.bz2&versionDescriptor%5BversionOptions%5D=0&versionDescriptor%5BversionType%5D=0&versionDescriptor%5Bversion%5D=conda_3.5_cuda90&download=true&resolveLfs=true&%24format=octetStream&api-version=5.0-preview.1) | [![Build status](https://dev.azure.com/pytorch/PyTorch/_apis/build/status/peterjc123.builder?branchName=master&jobName=Windows_CUDA90_Conda_Build&configuration=PY3.6)](https://dev.azure.com/peterjc123/46ef0f8e-f34b-4a52-b6e0-c1000caf14a1/_apis/git/repositories/49b30ad3-8e9b-43ff-bbc8-87179a36d246/Items?path=%2Fconda%2Fpytorch-nightly-1.0-py3.6_cuda90_cudnn7he774522_1.tar.bz2&versionDescriptor%5BversionOptions%5D=0&versionDescriptor%5BversionType%5D=0&versionDescriptor%5Bversion%5D=conda_3.6_cuda90&download=true&resolveLfs=true&%24format=octetStream&api-version=5.0-preview.1) | [![Build status](https://dev.azure.com/pytorch/PyTorch/_apis/build/status/peterjc123.builder?branchName=master&jobName=Windows_CUDA90_Conda_Build&configuration=PY3.7)](https://dev.azure.com/peterjc123/46ef0f8e-f34b-4a52-b6e0-c1000caf14a1/_apis/git/repositories/49b30ad3-8e9b-43ff-bbc8-87179a36d246/Items?path=%2Fconda%2Fpytorch-nightly-1.0-py3.7_cuda90_cudnn7he774522_1.tar.bz2&versionDescriptor%5BversionOptions%5D=0&versionDescriptor%5BversionType%5D=0&versionDescriptor%5Bversion%5D=conda_3.7_cuda90&download=true&resolveLfs=true&%24format=octetStream&api-version=5.0-preview.1) |
 | Windows x64 CUDA 9.0 (Wheels)  | [![Build status](https://dev.azure.com/pytorch/PyTorch/_apis/build/status/peterjc123.builder?branchName=master&jobName=Windows_CUDA90_Wheels_Build&configuration=PY3.5)](https://dev.azure.com/peterjc123/46ef0f8e-f34b-4a52-b6e0-c1000caf14a1/_apis/git/repositories/49b30ad3-8e9b-43ff-bbc8-87179a36d246/Items?path=%2Fwheels%2Fcuda90%2Ftorch-1.0-cp35-cp35m-win_amd64.whl&versionDescriptor%5BversionOptions%5D=0&versionDescriptor%5BversionType%5D=0&versionDescriptor%5Bversion%5D=wheels_3.5_cuda90&download=true&resolveLfs=true&%24format=octetStream&api-version=5.0-preview.1) | [![Build status](https://dev.azure.com/pytorch/PyTorch/_apis/build/status/peterjc123.builder?branchName=master&jobName=Windows_CUDA90_Wheels_Build&configuration=PY3.6)](https://dev.azure.com/peterjc123/46ef0f8e-f34b-4a52-b6e0-c1000caf14a1/_apis/git/repositories/49b30ad3-8e9b-43ff-bbc8-87179a36d246/Items?path=%2Fwheels%2Fcuda90%2Ftorch-1.0-cp36-cp36m-win_amd64.whl&versionDescriptor%5BversionOptions%5D=0&versionDescriptor%5BversionType%5D=0&versionDescriptor%5Bversion%5D=wheels_3.6_cuda90&download=true&resolveLfs=true&%24format=octetStream&api-version=5.0-preview.1) | [![Build status](https://dev.azure.com/pytorch/PyTorch/_apis/build/status/peterjc123.builder?branchName=master&jobName=Windows_CUDA90_Wheels_Build&configuration=PY3.7)](https://dev.azure.com/peterjc123/46ef0f8e-f34b-4a52-b6e0-c1000caf14a1/_apis/git/repositories/49b30ad3-8e9b-43ff-bbc8-87179a36d246/Items?path=%2Fwheels%2Fcuda90%2Ftorch-1.0-cp37-cp37m-win_amd64.whl&versionDescriptor%5BversionOptions%5D=0&versionDescriptor%5BversionType%5D=0&versionDescriptor%5Bversion%5D=wheels_3.7_cuda90&download=true&resolveLfs=true&%24format=octetStream&api-version=5.0-preview.1) |
 
+And here are the libtorch packages. Only shared build is supported now.
+
+| Configuration / LibTorch version | Shared library                                               |
+| -------------------------------- | ------------------------------------------------------------ |
+| Windows x64 CPU                  | [![Build status](https://dev.azure.com/pytorch/PyTorch/_apis/build/status/peterjc123.builder?branchName=master&jobName=Windows_CPU_Wheels_Build&configuration=LIBTORCH)](https://dev.azure.com/peterjc123/46ef0f8e-f34b-4a52-b6e0-c1000caf14a1/_apis/git/repositories/49b30ad3-8e9b-43ff-bbc8-87179a36d246/Items?path=%2Fwheels%2Fcpu%2Flibtorch-shared-with-deps-1.0.zip&versionDescriptor%5BversionOptions%5D=0&versionDescriptor%5BversionType%5D=0&versionDescriptor%5Bversion%5D=wheels_3&download=true&resolveLfs=true&%24format=octetStream&api-version=5.0-preview.1) |
+| Windows x64 CUDA 9.0             | [![Build status](https://dev.azure.com/pytorch/PyTorch/_apis/build/status/peterjc123.builder?branchName=master&jobName=Windows_CUDA90_Wheels_Build&configuration=LIBTORCH)](https://dev.azure.com/peterjc123/46ef0f8e-f34b-4a52-b6e0-c1000caf14a1/_apis/git/repositories/49b30ad3-8e9b-43ff-bbc8-87179a36d246/Items?path=%2Fwheels%2Fcuda90%2Flibtorch-shared-with-deps-1.0.zip&versionDescriptor%5BversionOptions%5D=0&versionDescriptor%5BversionType%5D=0&versionDescriptor%5Bversion%5D=wheels_3_cuda90&download=true&resolveLfs=true&%24format=octetStream&api-version=5.0-preview.1) |
+
 ## How to find the package and install?
 
 You can first click the icon to get to the main page of the CI system for a certain package. You can see a few jobs under that page. There're several variables that is used to distinguish the packages.
@@ -125,6 +132,9 @@ set SKIP_VS_VER_CHECK=1
 # If you want to skip prompt for building for Python 2.7 (Not recommended)
 set SKIP_PY_VER_CHECK=1
 
+# If you want to build for LibTorch
+set BUILD_PYTHONLESS=1
+
 ################################################################################
 # These are the available scripts that you can use.
 ################################################################################

internal/setup.bat

@@ -54,8 +54,37 @@ pushd %SRC_DIR%
 IF NOT exist "setup.py" (
     cd pytorch
 )
+
+if "%BUILD_PYTHONLESS%" == "" goto pytorch else goto libtorch
+
+:libtorch
+set VARIANT=shared-with-deps
+
+mkdir libtorch
+set "CMAKE_INSTALL_PREFIX=%CD%\libtorch"
+
+mkdir build
+pushd build
+python ../tools/build_libtorch.py
+popd
+
+IF ERRORLEVEL 1 goto no
+IF NOT ERRORLEVEL 0 goto no
+
+move /Y libtorch\bin\*.dll libtorch\lib\
+
+echo LibTorch is built in %CD%\libtorch.
+
+goto build_end
+
+:pytorch
 python setup.py install
 
+:build_end
+
+IF ERRORLEVEL 1 goto no
+IF NOT ERRORLEVEL 0 goto no
+
 goto :eof
 
 :no