Add main.cpp back, and invoke llama_main from it

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

Author: tarruda

CMakeLists.txt

@@ -213,6 +213,7 @@ endif()
 #
 
 add_executable(llama
+               main.cpp
                llama.cpp)
 
 add_executable(quantize quantize.cpp)

Makefile

@@ -223,11 +223,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)
 	@echo "\x1b[36mrun ./main -h for help\x1b[0m"
 
 quantize: quantize.cpp ggml.o utils.o

llama.cpp

@@ -849,7 +849,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);
+}