commit 538b15d13c368b748f0b3d1bd4984a34b20d591a
author Tongzhou Wang <tongzhou.wang.1994@gmail.com> Thu Aug 02 12:49:30 2018 -0700
committer Facebook Github Bot <facebook-github-bot@users.noreply.github.com> Thu Aug 02 12:54:14 2018 -0700
tree 1cb70db7d2ad2fcdc36fd89e86ac8bcad0f2a797
parent 9e85a7a9de6e974922347e8a44c0b55bf044cffb

Use PYTORCH_PYTHON to call generate_code.py (#10171)

Summary:
Probably fixes https://github.com/pytorch/pytorch/issues/8373#issuecomment-409994847
Pull Request resolved: https://github.com/pytorch/pytorch/pull/10171

Differential Revision: D9135607

Pulled By: SsnL

fbshipit-source-id: 72f535875658c857621e41fd25c2174052714557

torch/CMakeLists.txt

diff --git a/torch/CMakeLists.txt b/torch/CMakeLists.txt
index d64073d..b41b789 100644
--- a/torch/CMakeLists.txt
+++ b/torch/CMakeLists.txt
@@ -46,6 +46,13 @@
   list(APPEND CMAKE_INSTALL_RPATH "${CMAKE_INSTALL_PREFIX}/lib")
 endif()
 
+# Get the correct Python executable
+if (DEFINED ENV{PYTORCH_PYTHON})
+  message(STATUS "Using python found in $ENV{PYTORCH_PYTHON}")
+  set(PYCMD "$ENV{PYTORCH_PYTHON}")
+else()
+  SET(PYCMD "python")
+endif()
 
 # Generate files
 set(TOOLS_PATH "${TORCH_SRC_DIR}/../tools")
@@ -79,7 +86,7 @@
   "${TORCH_SRC_DIR}/csrc/jit/generated/register_aten_ops.cpp"
   "${TORCH_SRC_DIR}/csrc/jit/generated/aten_interned_strings.h"
   COMMAND
-  python tools/setup_helpers/generate_code.py
+  ${PYCMD} tools/setup_helpers/generate_code.py
     --declarations-path "${CMAKE_BINARY_DIR}/aten/src/ATen/Declarations.yaml"
     --nn-path "aten/src/"
   DEPENDS