Added the initial version of NGraph Plugin dynamic library.

src/CMakeLists.txt

@@ -21,6 +21,7 @@ set (SRC
     tree.cpp
     util.cpp
     log.cpp
+    ngraph.cpp
 
     transformers/axes.cpp
     transformers/exop.cpp
@@ -40,6 +41,12 @@ include_directories(
 
 add_library(ngraph SHARED ${SRC})
 
+if ( APPLE )
+    set_property( TARGET ngraph PROPERTY PREFIX "lib" )
+    set_property( TARGET ngraph PROPERTY OUTPUT_NAME "ngraph.so" )
+    set_property( TARGET ngraph PROPERTY SUFFIX "" )
+endif()
+
 #-----------------------------------------------------------------------------------------------
 # Installation logic...
 #-----------------------------------------------------------------------------------------------

src/ngraph.cpp

+// ----------------------------------------------------------------------------
+// Copyright 2017 Nervana Systems Inc.
+// 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
+// ----------------------------------------------------------------------------
+#include "ngraph.hpp"
+#include "log.hpp"
+
+NGraph* create_plugin()
+{
+    return new NGraph();
+}
+
+void destroy_plugin(NGraph* pObj)
+{
+    delete pObj;
+}
+
+void NGraph::add_params( const std::vector<std::string>& paramList )
+{
+    INFO << "Adding parameters";
+    m_params.insert(m_params.end(), paramList.begin(), paramList.end());    
+}

src/ngraph.hpp

+#pragma once
+// ----------------------------------------------------------------------------
+// Copyright 2017 Nervana Systems Inc.
+// 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
+// ----------------------------------------------------------------------------
+
+#include <string>
+#include <vector>
+
+class NGraph
+{
+public:
+    void add_params(const std::vector<std::string>& paramList);
+    const std::vector<std::string>& get_params() const { return m_params; }
+    std::string get_name() const { return "NGraph Plugin"; }
+
+private:
+    std::vector<std::string> m_params;
+};
+
+// Factory methods
+extern "C" NGraph* create_plugin();
+extern "C" void destroy_plugin(NGraph* pObj);
+
+// FUnction pointers to the factory methods
+typedef NGraph* (*CreatePluginPfn)();
+typedef void    (*DestroyPluginPfn)(NGraph*);

test/CMakeLists.txt

@@ -31,6 +31,7 @@ set (SRC
     uuid.cpp
     names.cpp
     strides.cpp
+    ngraph.cpp
 )
 
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")

test/main.cpp

@@ -16,6 +16,7 @@
 #include <chrono>
 
 #include "gtest/gtest.h"
+#include "log.hpp"
 
 using namespace std;
 

test/ngraph.cpp

+// ----------------------------------------------------------------------------
+// Copyright 2017 Nervana Systems Inc.
+// 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
+// ----------------------------------------------------------------------------
+
+#include <vector>
+#include <string>
+#include <sstream>
+#include <memory>
+#include <dlfcn.h>
+
+#include "gtest/gtest.h"
+#include "ngraph.hpp"
+#include "log.hpp"
+
+using namespace std;
+
+TEST(NGraph, loadTest)
+{
+    // load the triangle library
+    void* pluginLib = dlopen("../src/libngraph.so", RTLD_LAZY);
+    if (!pluginLib)
+    {
+        std::cerr << "Cannot load library: " << dlerror() << '\n';
+        ASSERT_FALSE(true);
+    }
+
+    // reset errors
+    dlerror();
+
+    // Get the symbols
+    auto createPfn = reinterpret_cast<CreatePluginPfn>(dlsym(pluginLib, "create_plugin"));
+    ASSERT_FALSE(createPfn == nullptr);
+
+    auto destroyPfn = reinterpret_cast<DestroyPluginPfn>(dlsym(pluginLib, "destroy_plugin"));
+    ASSERT_FALSE(destroyPfn == nullptr);
+
+    NGraph* pluginObj = createPfn();
+
+    INFO << "Call a method on the Object";
+    ASSERT_EQ("NGraph Plugin", pluginObj->get_name());
+    INFO << "Plugin Name: " << pluginObj->get_name();
+
+    // Add some parameters
+    const vector<string> TEST_PARAMS = {"param-1", "param-2", "param-3"};
+
+    pluginObj->add_params( TEST_PARAMS );
+
+    // Get the list of params
+    auto& storedParams = pluginObj->get_params();
+    EXPECT_EQ( TEST_PARAMS.size(), storedParams.size() );
+    for (int i = 0; i < TEST_PARAMS.size(); i++)
+    {
+        EXPECT_EQ( TEST_PARAMS[i], storedParams[i] );
+    }
+
+    INFO << "Destroy the Plugin Object";
+    destroyPfn(pluginObj);
+
+    dlclose(pluginLib);
+}