update CMake

Kaffaljidhmah2 · Mar 21, 2019 · 202009d · 202009d
1 parent 3317e35
commit 202009d
Show file tree

Hide file tree

Showing 5 changed files with 31 additions and 11 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -0,0 +1,15 @@
+cmake_minimum_required(VERSION 3.2)
+
+PROJECT(ts)
+
+INCLUDE_DIRECTORIES(
+include
+)
+
+AUX_SOURCE_DIRECTORY(src DIR_SRCS)
+
+SET(TMP
+${DIR_SRCS}
+)
+
+ADD_EXECUTABLE(${PROJECT_NAME} ${TMP})
diff --git a/variable.cpp → include/tensor.tpp b/variable.cpp → include/tensor.tpp
@@ -1,4 +1,4 @@
-#include "variable.hpp"
+#include "variable.h"
 
 namespace dlframework{
 

diff --git a/variable.hpp → include/variable.h b/variable.hpp → include/variable.h
@@ -36,4 +36,6 @@ namespace dlframework{
 
 }// end dlframework
 
+#include <tensor.tpp>
+
 #endif
diff --git a/src/test.cpp b/src/test.cpp
@@ -0,0 +1,13 @@
+#include "variable.h"
+#include <iostream>
+
+using namespace std;
+using namespace dlframework;
+
+int main()
+{
+ Tensor<int> a({2,3,5});
+ cout<<a.dim<<endl;
+ cout<<a({1,2,3})<<endl;
+
+}
diff --git a/test.cpp b/test.cpp