From 72e66f5fa324eb2b5a38e3c69de30cc3d7a128fe Mon Sep 17 00:00:00 2001 From: Christian Goll Date: Wed, 12 Jun 2019 11:10:04 +0200 Subject: [PATCH 2/2] honor PSIMD env --- tools/build_pytorch_libs.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tools/build_pytorch_libs.py b/tools/build_pytorch_libs.py index 257e9f2..3668b91 100644 --- a/tools/build_pytorch_libs.py +++ b/tools/build_pytorch_libs.py @@ -206,7 +206,8 @@ def run_cmake(version, USE_REDIS=os.getenv('USE_REDIS'), USE_GLOG=os.getenv('USE_GLOG'), USE_GFLAGS=os.getenv('USE_GFLAGS'), - WERROR=os.getenv('WERROR')) + WERROR=os.getenv('WERROR'), + PSIMD_SOURCE_DIR=os.getenv('PSIMD_SOURCE_DIR')) if USE_GLOO_IBVERBS: cmake_defines(cmake_args, USE_IBVERBS="1", USE_GLOO_IBVERBS="1") -- 2.16.4