This repository has been archived by the owner on Jan 3, 2023. It is now read-only.
/
ngraph-tf.cmake
65 lines (57 loc) · 2.56 KB
/
ngraph-tf.cmake
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# ******************************************************************************
# Copyright 2018-2019 Intel Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ******************************************************************************
include(ExternalProject)
set(EXTERNAL_NGRAPH_INSTALL_DIR ${EXTERNAL_INSTALL_DIR})
set(NGRAPH_TF_CMAKE_PREFIX ext_ngraph_tf)
SET(NGRAPH_TF_REPO_URL https://github.com/NervanaSystems/ngraph-tf.git)
SET(NGRAPH_TF_GIT_LABEL v0.12.0-rc0)
SET(NGRAPH_TF_SRC_DIR ${CMAKE_BINARY_DIR}/${NGRAPH_TF_CMAKE_PREFIX}/src/${NGRAPH_TF_CMAKE_PREFIX})
SET(NGRAPH_TF_BUILD_DIR ${NGRAPH_TF_SRC_DIR}/build)
SET(NGRAPH_TF_ARTIFACTS_DIR ${NGRAPH_TF_BUILD_DIR}/artifacts)
SET(NGRAPH_TF_VENV_DIR ${NGRAPH_TF_BUILD_DIR}/venv-tf-py3)
SET(NGRAPH_TF_VENV_LIB_DIR ${NGRAPH_TF_VENV_DIR}/lib/${PYTHON_VENV_VERSION}/site-packages/ngraph_bridge)
SET(NGRAPH_TF_INCLUDE_DIR ${NGRAPH_TF_ARTIFACTS_DIR}/include)
SET(NGRAPH_TF_LIB_DIR ${NGRAPH_TF_ARTIFACTS_DIR}/lib)
SET(NGRAPH_TEST_UTIL_INCLUDE_DIR ${NGRAPH_TF_BUILD_DIR}/ngraph/test)
if (USE_PREBUILT_BINARIES)
message(STATUS "Using prebuilt ng/ng-tf/tf binaries")
ExternalProject_Add(
ext_ngraph_tf
GIT_REPOSITORY ${NGRAPH_TF_REPO_URL}
GIT_TAG ${NGRAPH_TF_GIT_LABEL}
PREFIX ${NGRAPH_TF_CMAKE_PREFIX}
UPDATE_COMMAND ""
CONFIGURE_COMMAND ""
BUILD_IN_SOURCE 1
BUILD_BYPRODUCTS ${NGRAPH_TF_CMAKE_PREFIX}
BUILD_COMMAND python3 ${NGRAPH_TF_SRC_DIR}/build_ngtf.py --use_prebuilt_binaries
INSTALL_COMMAND ln -fs ${NGRAPH_TF_VENV_DIR} ${EXTERNAL_INSTALL_DIR}
)
else()
message(STATUS "Rebuilding ng/ng-tf/tf binaries")
ExternalProject_Add(
ext_ngraph_tf
GIT_REPOSITORY ${NGRAPH_TF_REPO_URL}
GIT_TAG ${NGRAPH_TF_GIT_LABEL}
PREFIX ${NGRAPH_TF_CMAKE_PREFIX}
UPDATE_COMMAND ""
CONFIGURE_COMMAND ""
BUILD_IN_SOURCE 1
BUILD_BYPRODUCTS ${NGRAPH_TF_CMAKE_PREFIX}
BUILD_COMMAND python3 ${NGRAPH_TF_SRC_DIR}/build_ngtf.py
INSTALL_COMMAND ln -fs ${NGRAPH_TF_VENV_DIR} ${EXTERNAL_INSTALL_DIR}
)
endif()