Christian Goll
0a90b0a626
* Distributed Model Parallel Training * Pruning functionalities have been added to PyTorch - New Features: * torch.optim.lr_scheduler now support “chaining.” * torch.distributed.rpc is a newly introduced package - full Changelog listed in relases file or under https://github.com/pytorch/pytorch/releases - added files: * skip-third-party-check.patch which is a patch to skip the check of disabled dependencies * QNNPACK-7d2a4e9931a82adc3814275b6219a03e24e36b4c.tar.gz which is part of pytorch but developed in different repo - removed patch files: * fix-build-options.patch * honor-PSIMD-env.patch * removed-some-tests.patch - Requires python-PeachPy on x86_64 only, as it is optional and available on x86_64 only OBS-URL: https://build.opensuse.org/package/show/science:machinelearning/python-torch?expand=0&rev=7
70 lines
2.9 KiB
Diff
70 lines
2.9 KiB
Diff
From 0f1b2fb9bd2b9ea9ae7c25b366c74dd9daf39439 Mon Sep 17 00:00:00 2001
|
|
From: Christian Goll <cgoll@suse.de>
|
|
Date: Thu, 20 Feb 2020 15:18:18 +0100
|
|
Subject: [PATCH] skip third party check Add the possibilty to skip check for
|
|
subdirs in third_party
|
|
|
|
---
|
|
setup.py | 37 +++++++++++++++++++++----------------
|
|
1 file changed, 21 insertions(+), 16 deletions(-)
|
|
|
|
diff --git a/setup.py b/setup.py
|
|
index 01fee0e3a5..4c3d86cee6 100644
|
|
--- a/setup.py
|
|
+++ b/setup.py
|
|
@@ -160,7 +160,9 @@
|
|
# USE_TBB
|
|
# enable TBB support
|
|
#
|
|
-
|
|
+# USE_SYSTEM_LIB
|
|
+# A comma seperated list of libraries for which should not be checked
|
|
+# under the directory third_party
|
|
from __future__ import print_function
|
|
from setuptools import setup, Extension, distutils, find_packages
|
|
from collections import defaultdict
|
|
@@ -287,22 +289,25 @@ def build_deps():
|
|
report('-- Building version ' + version)
|
|
|
|
def check_file(f):
|
|
- if not os.path.exists(f):
|
|
- report("Could not find {}".format(f))
|
|
- report("Did you run 'git submodule update --init --recursive'?")
|
|
- sys.exit(1)
|
|
-
|
|
- check_file(os.path.join(third_party_path, "gloo", "CMakeLists.txt"))
|
|
- check_file(os.path.join(third_party_path, "pybind11", "CMakeLists.txt"))
|
|
- check_file(os.path.join(third_party_path, 'cpuinfo', 'CMakeLists.txt'))
|
|
- check_file(os.path.join(third_party_path, 'tbb', 'Makefile'))
|
|
- check_file(os.path.join(third_party_path, 'onnx', 'CMakeLists.txt'))
|
|
- check_file(os.path.join(third_party_path, 'foxi', 'CMakeLists.txt'))
|
|
- check_file(os.path.join(third_party_path, 'QNNPACK', 'CMakeLists.txt'))
|
|
- check_file(os.path.join(third_party_path, 'fbgemm', 'CMakeLists.txt'))
|
|
- check_file(os.path.join(third_party_path, 'fbgemm', 'third_party',
|
|
+ if os.path.dirname(f) in str(os.getenv('USE_SYSTEM_LIB')):
|
|
+ report("Not checking for {} in third_party".format(f))
|
|
+ else:
|
|
+ if not os.path.exists(os.path.join(third_party_path,f)):
|
|
+ report("Could not find {}".format(os.path.join(third_party_path,f)))
|
|
+ report("Did you run 'git submodule update --init --recursive'?")
|
|
+ sys.exit(1)
|
|
+
|
|
+ check_file(os.path.join("gloo", "CMakeLists.txt"))
|
|
+ check_file(os.path.join("pybind11", "CMakeLists.txt"))
|
|
+ check_file(os.path.join('cpuinfo', 'CMakeLists.txt'))
|
|
+ check_file(os.path.join('tbb', 'Makefile'))
|
|
+ check_file(os.path.join('onnx', 'CMakeLists.txt'))
|
|
+ check_file(os.path.join('foxi', 'CMakeLists.txt'))
|
|
+ check_file(os.path.join('QNNPACK', 'CMakeLists.txt'))
|
|
+ check_file(os.path.join('fbgemm', 'CMakeLists.txt'))
|
|
+ check_file(os.path.join('fbgemm', 'third_party',
|
|
'asmjit', 'CMakeLists.txt'))
|
|
- check_file(os.path.join(third_party_path, 'onnx', 'third_party',
|
|
+ check_file(os.path.join('onnx', 'third_party',
|
|
'benchmark', 'CMakeLists.txt'))
|
|
|
|
check_pydep('yaml', 'pyyaml')
|
|
--
|
|
2.25.0
|
|
|