Christian Goll
ba0a6629a5
version, so these are plain .so files - THNN.h and THCUNN.h are interpred by the python and can so not be part of the devel package OBS-URL: https://build.opensuse.org/package/show/science:machinelearning/python-torch?expand=0&rev=1
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
|
|
|