add test folder and cmakelists.txt for test (#4)

Author: EeyoreLee

.gitignore

@@ -31,4 +31,5 @@
 *.out
 *.app
 
-data/
+data/
+build/

.vscode/c_cpp_properties.json

@@ -4,10 +4,10 @@
             "name": "Linux",
             "includePath": [
                 "${workspaceFolder}/**",
-                "${workspaceFolder}/ggml/**"
+                "${workspaceFolder}/ggml/**",
+                "${workspaceFolder}/src/**"
             ],
             "defines": [],
-            "compilerPath": "/usr/bin/gcc",
             "cStandard": "c17",
             "cppStandard": "gnu++17",
             "intelliSenseMode": "linux-gcc-x64"

.vscode/settings.json

@@ -9,4 +9,11 @@
         "editor.formatOnType": false
     },
     "yapf.importStrategy": "useBundled",
+    "cmake.sourceDirectory": [
+        "/media/E/lichunyu/bert.cpp",
+        "/media/E/lichunyu/bert.cpp/cmake",
+        "/media/E/lichunyu/bert.cpp/test/test_model_load"
+    ],
+    "C_Cpp.default.compilerPath": "/usr/bin/g++-11",
+    "cmake.configureOnOpen": true,
 }

cmake/CMakeLists.txt

@@ -0,0 +1 @@
+cmake_minimum_required(VERSION 3.12)

src/bert.cpp

@@ -5,9 +5,11 @@
 #include <fstream>
 
 int bert_predict() {
+    return 0;
 };
 
 std::vector<int> bert_batch_predict() {
+    return std::vector<int>{1, 2};
 };
 
 bool bert_model_load(const std::string &fname, bert_model &model)
@@ -82,4 +84,5 @@ bool bert_model_load(const std::string &fname, bert_model &model)
         return false;
     }
     }
+    return true;
 };

test/test_model_load/CMakeLists.txt

@@ -0,0 +1,10 @@
+cmake_minimum_required(VERSION 3.12)
+project(test_model_load)
+
+add_subdirectory(../../ggml ${CMAKE_CURRENT_BINARY_DIR}/ggml)
+
+include_directories(../../src)
+
+add_executable(test_model_load test_model_load.cpp ../../src/bert.cpp)
+
+target_link_libraries(test_model_load ggml)

test/test_model_load/test_model_load.cpp

@@ -0,0 +1,14 @@
+#include "bert.h"
+
+#include <string>
+
+int main(int argc, char **argv)
+{
+    bert_model model;
+    std::string fname = "/media/E/lichunyu/bert.cpp/data/ggml-model/ggml_model.gguf";
+    if (!bert_model_load(fname, model))
+    {
+        fprintf(stderr, "%s: failed to load model from %s", __func__, fname.c_str());
+    }
+    return 0;
+}