Commit e9d9732e authored by Adam Procter's avatar Adam Procter

Merge remote-tracking branch 'origin/master' into r0.10

parents 88986560 281c8ea1
The MPL 2.0 license used by the eigen library used by this ngraph core
component requires distribution of the following information:
Eigen source code can be viewed or downloaded from here:
http://eigen.tuxfamily.org
...@@ -22,7 +22,7 @@ import os ...@@ -22,7 +22,7 @@ import os
import distutils.ccompiler import distutils.ccompiler
__version__ = os.environ.get('NGRAPH_VERSION', '0.0.0-dev') __version__ = os.environ.get('NGRAPH_VERSION', '0.0.0-dev')
PYNGRAPH_SOURCE_DIR = os.path.abspath(os.path.dirname(__file__)) PYNGRAPH_ROOT_DIR = os.path.abspath(os.path.dirname(__file__))
NGRAPH_DEFAULT_INSTALL_DIR = os.environ.get('HOME') NGRAPH_DEFAULT_INSTALL_DIR = os.environ.get('HOME')
NGRAPH_ONNX_IMPORT_ENABLE = os.environ.get('NGRAPH_ONNX_IMPORT_ENABLE') NGRAPH_ONNX_IMPORT_ENABLE = os.environ.get('NGRAPH_ONNX_IMPORT_ENABLE')
...@@ -50,7 +50,7 @@ def find_pybind_headers_dir(): ...@@ -50,7 +50,7 @@ def find_pybind_headers_dir():
if os.environ.get('PYBIND_HEADERS_PATH'): if os.environ.get('PYBIND_HEADERS_PATH'):
pybind_headers_dir = os.environ.get('PYBIND_HEADERS_PATH') pybind_headers_dir = os.environ.get('PYBIND_HEADERS_PATH')
else: else:
pybind_headers_dir = os.path.join(PYNGRAPH_SOURCE_DIR, 'pybind11') pybind_headers_dir = os.path.join(PYNGRAPH_ROOT_DIR, 'pybind11')
found = os.path.exists(os.path.join(pybind_headers_dir, 'include/pybind11')) found = os.path.exists(os.path.join(pybind_headers_dir, 'include/pybind11'))
if not found: if not found:
...@@ -233,13 +233,13 @@ sources = [ ...@@ -233,13 +233,13 @@ sources = [
] ]
package_dir = { package_dir = {
'ngraph': PYNGRAPH_SOURCE_DIR + "/ngraph", 'ngraph': PYNGRAPH_ROOT_DIR + "/ngraph",
'ngraph.utils': PYNGRAPH_SOURCE_DIR + "/ngraph/utils", 'ngraph.utils': PYNGRAPH_ROOT_DIR + "/ngraph/utils",
'ngraph.impl': PYNGRAPH_SOURCE_DIR + "/ngraph/impl", 'ngraph.impl': PYNGRAPH_ROOT_DIR + "/ngraph/impl",
'ngraph.impl.op': PYNGRAPH_SOURCE_DIR + "/ngraph/impl/op", 'ngraph.impl.op': PYNGRAPH_ROOT_DIR + "/ngraph/impl/op",
'ngraph.impl.op.util': PYNGRAPH_SOURCE_DIR + "/ngraph/impl/op/util", 'ngraph.impl.op.util': PYNGRAPH_ROOT_DIR + "/ngraph/impl/op/util",
'ngraph.impl.passes': PYNGRAPH_SOURCE_DIR + "/ngraph/impl/passes", 'ngraph.impl.passes': PYNGRAPH_ROOT_DIR + "/ngraph/impl/passes",
'ngraph.impl.runtime': PYNGRAPH_SOURCE_DIR + "/ngraph/impl/runtime", 'ngraph.impl.runtime': PYNGRAPH_ROOT_DIR + "/ngraph/impl/runtime",
} }
packages = [ packages = [
'ngraph', 'ngraph',
...@@ -251,9 +251,9 @@ packages = [ ...@@ -251,9 +251,9 @@ packages = [
'ngraph.impl.runtime', 'ngraph.impl.runtime',
] ]
sources = [PYNGRAPH_SOURCE_DIR + "/" + source for source in sources] sources = [PYNGRAPH_ROOT_DIR + "/" + source for source in sources]
include_dirs = [PYNGRAPH_SOURCE_DIR, NGRAPH_CPP_INCLUDE_DIR, PYBIND11_INCLUDE_DIR] include_dirs = [PYNGRAPH_ROOT_DIR, NGRAPH_CPP_INCLUDE_DIR, PYBIND11_INCLUDE_DIR]
library_dirs = [NGRAPH_CPP_LIBRARY_DIR] library_dirs = [NGRAPH_CPP_LIBRARY_DIR]
...@@ -274,13 +274,13 @@ data_files = [ ...@@ -274,13 +274,13 @@ data_files = [
( (
'licenses', 'licenses',
[ [
PYNGRAPH_SOURCE_DIR + "/../licenses/" + license PYNGRAPH_ROOT_DIR + "/../licenses/" + license
for license in os.listdir(PYNGRAPH_SOURCE_DIR + "/../licenses") for license in os.listdir(PYNGRAPH_ROOT_DIR + "/../licenses")
], ],
), ),
( (
'', '',
[PYNGRAPH_SOURCE_DIR + "/../LICENSE"], [PYNGRAPH_ROOT_DIR + "/../LICENSE"],
) )
] ]
...@@ -302,10 +302,10 @@ if NGRAPH_ONNX_IMPORT_ENABLE == 'TRUE': ...@@ -302,10 +302,10 @@ if NGRAPH_ONNX_IMPORT_ENABLE == 'TRUE':
'pyngraph/pyngraph_onnx_import.cpp', 'pyngraph/pyngraph_onnx_import.cpp',
'pyngraph/onnx_import/onnx_import.cpp', 'pyngraph/onnx_import/onnx_import.cpp',
] ]
onnx_sources = [PYNGRAPH_SOURCE_DIR + "/" + source for source in onnx_sources] onnx_sources = [PYNGRAPH_ROOT_DIR + "/" + source for source in onnx_sources]
package_dir['ngraph.impl.onnx_import'] = ( package_dir['ngraph.impl.onnx_import'] = (
PYNGRAPH_SOURCE_DIR + "/ngraph/impl/onnx_import" PYNGRAPH_ROOT_DIR + "/ngraph/impl/onnx_import"
) )
packages.append('ngraph.impl.onnx_import') packages.append('ngraph.impl.onnx_import')
...@@ -360,17 +360,17 @@ class BuildExt(build_ext): ...@@ -360,17 +360,17 @@ class BuildExt(build_ext):
build_ext.build_extensions(self) build_ext.build_extensions(self)
with open(os.path.join(PYNGRAPH_SOURCE_DIR, 'requirements.txt')) as req: with open(os.path.join(PYNGRAPH_ROOT_DIR, 'requirements.txt')) as req:
requirements = req.read().splitlines() requirements = req.read().splitlines()
setup( setup(
name='ngraph-core', name='ngraph-core',
description=open(os.path.join(PYNGRAPH_ROOT_DIR, 'README.md')).read(),
version=__version__, version=__version__,
author='Intel', author='Intel',
author_email='intelnervana@intel.com', author_email='intelnervana@intel.com',
url='https://ai.intel.com/', url='https://ai.intel.com/',
license='License :: OSI Approved :: Apache Software License', license='License :: OSI Approved :: Apache Software License',
description='Python API for nGraph',
long_description='', long_description='',
ext_modules=ext_modules, ext_modules=ext_modules,
package_dir=package_dir, package_dir=package_dir,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment