27 lines
860 B
Diff
27 lines
860 B
Diff
|
From 72e66f5fa324eb2b5a38e3c69de30cc3d7a128fe Mon Sep 17 00:00:00 2001
|
||
|
|
From: Christian Goll <cgoll@suse.de>
|
||
|
|
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
|
||
|
|
|