diff options
| -rw-r--r-- | tools/build_pytorch_libs.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/tools/build_pytorch_libs.py b/tools/build_pytorch_libs.py index 6c163f5799..aa0ed89cb8 100644 --- a/tools/build_pytorch_libs.py +++ b/tools/build_pytorch_libs.py @@ -112,6 +112,13 @@ def create_build_env(): if IS_WINDOWS: my_env = overlay_windows_vcvars(my_env) + # When using Ninja under Windows, the gcc toolchain will be chosen as default. + # But it should be set to MSVC as the user's first choice. + if USE_NINJA: + cc = my_env.get('CC', 'cl') + cxx = my_env.get('CXX', 'cl') + my_env['CC'] = cc + my_env['CXX'] = cxx return my_env @@ -258,7 +265,7 @@ def build_caffe2(version, check_call(['cmake', '--build', '.', '--target', 'install', '--config', build_type, '--', '-j', str(j)], cwd=build_dir, env=my_env) else: - check_call(['msbuild', 'INSTALL.vcxproj', '/p:Configuration={}'.format(build_type)], + check_call(['msbuild', 'INSTALL.vcxproj', '/p:Configuration={} /maxcpucount:{}'.format(build_type, j)], cwd=build_dir, env=my_env) else: if USE_NINJA: |