add lsflow

lightseq/csrc/lsflow/CMakeLists.txt

@@ -0,0 +1,26 @@
+cmake_minimum_required(VERSION 3.10)
+project(LightseqProtoType LANGUAGES CXX)
+find_package(Threads REQUIRED)
+set(CMAKE_CXX_STANDARD 14)
+
+include_directories(includes)
+
+add_definitions(-DDEBUG)
+
+add_executable(lsflow_bin example/main.cpp)
+
+add_library(node SHARED node.cpp)
+
+add_library(context SHARED context.cpp)
+
+add_library(manager SHARED manager.cpp)
+
+add_library(layer SHARED layer.cpp)
+
+add_library(tensor SHARED tensor.cpp)
+
+target_link_libraries(lsflow_bin PUBLIC node)
+target_link_libraries(lsflow_bin PUBLIC context)
+target_link_libraries(lsflow_bin PUBLIC manager)
+target_link_libraries(lsflow_bin PUBLIC layer)
+target_link_libraries(lsflow_bin PUBLIC tensor)

lightseq/csrc/lsflow/README.md

@@ -0,0 +1 @@
+LsFlow is a extremely clean implement of computation graph.

lightseq/csrc/lsflow/build.sh

@@ -0,0 +1,5 @@
+if [ ! -d 'build' ]; then
+    mkdir build
+fi
+
+cd build && cmake .. && make -j${nproc}

lightseq/csrc/core/context.cpp → lightseq/csrc/lsflow/context.cpp

@@ -50,16 +50,16 @@ void Context::build() {
   }
 
   free(temporary_buffer_);
-  memory_manager_->calculate_buffer_();
+  _mm_ptr->calculate_buffer_();
   _built = true;
 
-  // thread_context_ptr.reset(nullptr);
-
 #ifdef DEBUG
   draw_all_context();
 #endif
 }
 
 void Context::draw_all_context() {}
 
+// thread_local ContextPtr thread_context_ptr = nullptr;
+
 }  // namespace lightseq

lightseq/csrc/lsflow/example/CMakeLists.txt

@@ -0,0 +1,9 @@
+add_executable(lsflow_bin test_example.cpp)
+
+include_directories(../includes)
+
+target_link_libraries(lsflow_bin PUBLIC node)
+target_link_libraries(lsflow_bin PUBLIC context)
+target_link_libraries(lsflow_bin PUBLIC manager)
+target_link_libraries(lsflow_bin PUBLIC layer)
+target_link_libraries(lsflow_bin PUBLIC tensor)