Merge pull request #434 from tensorflow/shrestha/tag_rc3

Shrestha/merge master

Author: Shrestha Malik

CMakeLists.txt

@@ -248,7 +248,7 @@ if (NOT USE_PRE_BUILT_NGRAPH)
     ExternalProject_Add(
         ext_ngraph
         GIT_REPOSITORY https://github.com/NervanaSystems/ngraph
-        GIT_TAG v0.27.1-rc.1
+        GIT_TAG v0.28.0-rc.0
         CMAKE_ARGS 
             -DNGRAPH_DISTRIBUTED_ENABLE=${NGRAPH_DISTRIBUTED_ENABLE}
             -DNGRAPH_INSTALL_PREFIX=${NGRAPH_ARTIFACTS_DIR}
@@ -275,6 +275,9 @@ if (NOT USE_PRE_BUILT_NGRAPH)
 endif()
 set(NGRAPH_INSTALL_DIR ${NGRAPH_ARTIFACTS_DIR})
 
+if(NGRAPH_BRIDGE_STATIC_LIB_ENABLE)
+    add_definitions(-DNGRAPH_BRIDGE_STATIC_LIB_ENABLE)
+endif()
 
 if(OS_VERSION STREQUAL "\"centos\"")
     set(LIB "lib64")

README.md

@@ -88,7 +88,7 @@ Once TensorFlow's dependencies are installed, clone the `ngraph-bridge` repo:
 
         git clone https://github.com/tensorflow/ngraph-bridge.git
         cd ngraph-bridge
-        git checkout v0.22.0-rc1
+        git checkout v0.22.0-rc2
 
 Run the following Python script to build TensorFlow, nGraph, and the bridge. Use Python 3.5:
 

bazel/BUILD

@@ -134,6 +134,7 @@ cc_library(
         "-Wno-strict-overflow",
         '-D NGRAPH_BRIDGE_STATIC_LIB_ENABLE',
         '-D NGRAPH_TF_USE_GRAPPLER_OPTIMIZER',
+        '-D NGRAPH_CPU_STATIC_LIB_ENABLE',
         "-I ngraph_bridge",
         "-I logging",
         "-I external/ngraph/src",

bazel/WORKSPACE

@@ -55,11 +55,11 @@ tf_workspace(path_prefix = "", tf_repo_name = "org_tensorflow")
 http_archive(
     name = "ngraph",
     build_file = "//:bazel/ngraph.BUILD",
-    sha256 = "903991dfdb63304bc08bebab68e0fcb3e62690763b886c42178f363a0beb00f4",
-    strip_prefix = "ngraph-0.27.1-rc.1",
+    sha256 = "be59da1e04840b3980d3bcba7925b99ac2007e37e3952ef7c77caedba9734654",
+    strip_prefix = "ngraph-0.28.0-rc.0",
     urls = [
-        "https://mirror.bazel.build/github.com/NervanaSystems/ngraph/archive/v0.27.1-rc.1.tar.gz",
-        "https://github.com/NervanaSystems/ngraph/archive/v0.27.1-rc.1.tar.gz"
+        "https://mirror.bazel.build/github.com/NervanaSystems/ngraph/archive/v0.28.0-rc.0.tar.gz",
+        "https://github.com/NervanaSystems/ngraph/archive/v0.28.0-rc.0.tar.gz"
     ],
 )
 

bazel/ngraph.BUILD

@@ -68,12 +68,13 @@ cc_library(
         "-fstack-protector-all",
         '-D SHARED_LIB_PREFIX=\\"lib\\"',
         '-D SHARED_LIB_SUFFIX=\\".so\\"',
-        '-D NGRAPH_VERSION=\\"v0.27.1-rc.1\\"',
+        '-D NGRAPH_VERSION=\\"v0.28.0-rc.0\\"',
         "-D NGRAPH_DEX_ONLY",
         '-D PROJECT_ROOT_DIR=\\"\\"',
         '-D NGRAPH_STATIC_LIB_ENABLE',
         '-D NGRAPH_DYNAMIC_COMPONENTS_ENABLE',
         '-D NGRAPH_ENABLE_CPU_CONV_AUTO',
+        '-D NGRAPH_USE_LEGACY_MKLDNN',
         "-march=native",
         "-mtune=native",
         "-Wall",
@@ -111,9 +112,10 @@ cc_library(
         "-fstack-protector-all",
         '-D SHARED_LIB_PREFIX=\\"lib\\"',
         '-D SHARED_LIB_SUFFIX=\\".so\\"',
-        '-D NGRAPH_VERSION=\\"v0.27.1-rc.1\\"',
+        '-D NGRAPH_VERSION=\\"v0.28.0-rc.0\\"',
         "-D NGRAPH_DEX_ONLY",
         '-D PROJECT_ROOT_DIR=\\"\\"',
+        '-D NGRAPH_USE_LEGACY_MKLDNN',
     ] + CXX_ABI,
     linkopts = [
         "-Wl,-z,noexecstack",
@@ -166,6 +168,7 @@ cc_library(
         "src/ngraph/runtime/cpu/builder/convert.cpp",
         "src/ngraph/runtime/cpu/builder/convert_layout.cpp",
         "src/ngraph/runtime/cpu/builder/convolution.cpp",
+        "src/ngraph/runtime/cpu/builder/cum_sum.cpp",
         "src/ngraph/runtime/cpu/builder/dot.cpp",
         "src/ngraph/runtime/cpu/builder/dropout.cpp",
         "src/ngraph/runtime/cpu/builder/embedding_lookup.cpp",
@@ -269,10 +272,11 @@ cc_library(
         "-fstack-protector-all",
         '-D SHARED_LIB_PREFIX=\\"lib\\"',
         '-D SHARED_LIB_SUFFIX=\\".so\\"',
-        '-D NGRAPH_VERSION=\\"v0.27.1-rc.1\\"',
+        '-D NGRAPH_VERSION=\\"v0.28.0-rc.0\\"',
         "-D NGRAPH_DEX_ONLY",
         '-D PROJECT_ROOT_DIR=\\"\\"',
         '-D NGRAPH_CPU_STATIC_LIB_ENABLE',
+        '-D NGRAPH_USE_LEGACY_MKLDNN=\\"TRUE\\"',
         "-march=native",
         "-mtune=native",
         "-Wall",

build_ngtf.py

@@ -53,7 +53,7 @@ def main():
     '''
 
     # Component versions
-    ngraph_version = "v0.27.1-rc.1"
+    ngraph_version = "v0.28.0-rc.0"
     tf_version = "v1.14.0"
 
     # Command line parser options
@@ -412,6 +412,7 @@ def main():
 
     if (arguments.use_ngraph_staticlibs):
         ngraph_tf_cmake_flags.extend(["-DNGRAPH_BRIDGE_STATIC_LIB_ENABLE=TRUE"])
+
     if (arguments.debug_build):
         ngraph_tf_cmake_flags.extend(["-DCMAKE_BUILD_TYPE=Debug"])
 

examples/cpp/hello_tf.cpp

@@ -35,16 +35,20 @@
 using namespace std;
 
 // Prints the available backends
-void PrintAvailableBackends() {
+int PrintAvailableBackends() {
   // Get the list of backends
   auto supported_backends =
       tensorflow::ngraph_bridge::BackendManager::GetSupportedBackendNames();
   vector<string> backends(supported_backends.begin(), supported_backends.end());
-
-  cout << "Available backends: " << endl;
+  if (backends.empty()) {
+    std::cout << "No backend available " << std::endl;
+    return -1;
+  }
+  std::cout << "Available backends: " << std::endl;
   for (auto& backend_name : backends) {
-    cout << "Backend: " << backend_name << std::endl;
+    std::cout << "Backend: " << backend_name << std::endl;
   }
+  return 0;
 }
 
 // Create a simple computation graph and run
@@ -116,13 +120,18 @@ void PrintVersion() {
                     ? std::string("Yes")
                     : std::string("No"))
             << std::endl;
-
-  PrintAvailableBackends();
+  std::cout << std::endl;
 }
 
 int main(int argc, char** argv) {
   PrintVersion();
 
+  // Print the avialable backends and if none are available
+  // error out
+  if (PrintAvailableBackends()) {
+    return -1;
+  }
+
   // Run a simple example
   RunSimpleNetworkExample();
 

examples/cpp/inference/CMakeLists.txt

@@ -61,6 +61,10 @@ if (APPLE)
     add_definitions(-DTEST_SINGLE_INSTANCE)
 endif()
 
+if(NGRAPH_BRIDGE_STATIC_LIB_ENABLE)
+    add_definitions(-DNGRAPH_BRIDGE_STATIC_LIB_ENABLE)
+endif()
+
 if(NGRAPH_BRIDGE_STATIC_LIB_ENABLE)
     target_link_libraries(${APP_NAME} 
         -Wl,--whole-archive

examples/cpp/inference/infer_multiple_networks.cc

@@ -164,6 +164,13 @@ int main(int argc, char** argv) {
     return -1;
   }
 
+// Register cpu backend for static linking
+// [TODO]: Revisit this to see if we can remove registering here and register
+// only in BackendManager.
+#if defined(NGRAPH_BRIDGE_STATIC_LIB_ENABLE)
+  ngraph_register_cpu_backend();
+#endif
+
   const char* backend = "CPU";
   if (SetNGraphBackend(backend) != tf::Status::OK()) {
     std::cout << "Error: Cannot set the backend: " << backend << std::endl;

examples/cpp/inference/infer_single_network.cc

@@ -160,6 +160,11 @@ int main(int argc, char** argv) {
     return -1;
   }
 
+// Register cpu backend for static linking
+#if defined(NGRAPH_BRIDGE_STATIC_LIB_ENABLE)
+  ngraph_register_cpu_backend();
+#endif
+
   const char* backend = "CPU";
   if (SetNGraphBackend(backend) != tf::Status::OK()) {
     std::cout << "Error: Cannot set the backend: " << backend << std::endl;