Add main.cpp back, and invoke llama_main from it

Signed-off-by: Thiago Padilha <[email protected]>

Author: tarruda

CMakeLists.txt

@@ -107,8 +107,9 @@ endif()
 # endif()
 
 add_executable(llama
-    llama.cpp
+    main.cpp
     utils.cpp
+    llama.cpp
     utils.h)
 
 add_executable(quantize

Makefile

@@ -188,11 +188,14 @@ ggml.o: ggml.c ggml.h
 utils.o: utils.cpp utils.h
 	$(CXX) $(CXXFLAGS) -c utils.cpp -o utils.o
 
+llama.o: llama.cpp llama.h
+	$(CXX) $(CXXFLAGS) -c llama.cpp -o llama.o
+
 clean:
 	rm -f *.o main quantize
 
-main: llama.cpp ggml.o utils.o
-	$(CXX) $(CXXFLAGS) llama.cpp ggml.o utils.o -o main $(LDFLAGS)
+main: main.cpp ggml.o utils.o llama.o
+	$(CXX) $(CXXFLAGS) main.cpp ggml.o utils.o llama.o -o main $(LDFLAGS)
 	./main -h
 
 quantize: quantize.cpp ggml.o utils.o

llama.cpp

@@ -731,7 +731,7 @@ const char * llama_print_system_info(void) {
     return s.c_str();
 }
 
-int main(int argc, char ** argv) {
+int llama_main(int argc, char ** argv) {
     ggml_time_init();
     const int64_t t_main_start_us = ggml_time_us();
 

llama.h

@@ -57,3 +57,5 @@ struct llama_model {
     struct ggml_context * ctx;
     std::map<std::string, struct ggml_tensor *> tensors;
 };
+
+int llama_main(int argc, char ** argv);

main.cpp

@@ -0,0 +1,5 @@
+#include "llama.h"
+
+int main(int argc, char ** argv) {
+    return llama_main(argc, argv);
+}