From 89d1fad7ee312be58cf1159f87f67d5de49b00b9 Mon Sep 17 00:00:00 2001
From: Justine Tunney <jtunney@gmail.com>
Date: Sat, 13 May 2023 21:16:03 -0700
Subject: [PATCH] Enable crash reports for radpajama executables

---
 third_party/radpajama/copy-gptneox.cc        | 3 +++
 third_party/radpajama/main-redpajama-chat.cc | 5 +++++
 third_party/radpajama/main-redpajama.cc      | 3 +++
 third_party/radpajama/quantize-gptneox.cc    | 3 +++
 third_party/radpajama/radpajama.mk           | 1 +
 5 files changed, 15 insertions(+)

diff --git a/third_party/radpajama/copy-gptneox.cc b/third_party/radpajama/copy-gptneox.cc
index 1050673c6..649e2ae7c 100644
--- a/third_party/radpajama/copy-gptneox.cc
+++ b/third_party/radpajama/copy-gptneox.cc
@@ -26,6 +26,7 @@
 │  SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                      │
 │                                                                              │
 ╚─────────────────────────────────────────────────────────────────────────────*/
+#include "libc/log/log.h"
 #include "third_party/ggml/ggml.h"
 #include "third_party/libcxx/cstdio"
 #include "third_party/libcxx/map"
@@ -47,6 +48,8 @@ static const std::map<std::string, enum gptneox_ftype> GPTNEOX_FTYPE_MAP = {
 //  ./quantize models/llama/ggml-model.bin models/llama/ggml-model-quant.bin type
 //
 int main(int argc, char ** argv) {
+    ShowCrashReports();
+
     ggjt_v1();
     ggml_time_init();
 
diff --git a/third_party/radpajama/main-redpajama-chat.cc b/third_party/radpajama/main-redpajama-chat.cc
index ff0e12422..37450de7e 100644
--- a/third_party/radpajama/main-redpajama-chat.cc
+++ b/third_party/radpajama/main-redpajama-chat.cc
@@ -31,6 +31,7 @@
 #include "libc/calls/struct/sigaction.h"
 #include "libc/calls/struct/siginfo.h"
 #include "libc/calls/weirdtypes.h"
+#include "libc/log/log.h"
 #include "libc/runtime/pathconf.h"
 #include "libc/runtime/runtime.h"
 #include "libc/runtime/sysconf.h"
@@ -84,6 +85,8 @@ int main(int argc, char ** argv) {
     gpt_params params;
     params.model = "./examples/redpajama/models/pythia/ggml-RedPajama-INCITE-Chat-3B-v1-f16.bin";
     
+    ShowCrashReports();
+
     if (gpt_params_parse(argc, argv, params) == false) {
         return 1;
     }
@@ -186,6 +189,8 @@ int main(int argc, char ** argv) {
         return 0;
     }
 
+    ShowCrashReports();
+
     // Always interactive for RedPajama chat model
     params.interactive = true;
     
diff --git a/third_party/radpajama/main-redpajama.cc b/third_party/radpajama/main-redpajama.cc
index f17ebf308..0aad11063 100644
--- a/third_party/radpajama/main-redpajama.cc
+++ b/third_party/radpajama/main-redpajama.cc
@@ -31,6 +31,7 @@
 #include "libc/calls/struct/sigaction.h"
 #include "libc/calls/struct/siginfo.h"
 #include "libc/calls/weirdtypes.h"
+#include "libc/log/log.h"
 #include "libc/runtime/pathconf.h"
 #include "libc/runtime/runtime.h"
 #include "libc/runtime/sysconf.h"
@@ -83,6 +84,8 @@ int main(int argc, char ** argv) {
     gpt_params params;
     params.model = "./examples/redpajama/models/pythia/ggml-RedPajama-INCITE-Instruct-3B-v1-f16.bin";
 
+    ShowCrashReports();
+
     if (gpt_params_parse(argc, argv, params) == false) {
         return 1;
     }
diff --git a/third_party/radpajama/quantize-gptneox.cc b/third_party/radpajama/quantize-gptneox.cc
index 4ed8cd6a7..4d68f8325 100644
--- a/third_party/radpajama/quantize-gptneox.cc
+++ b/third_party/radpajama/quantize-gptneox.cc
@@ -26,6 +26,7 @@
 │  SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                      │
 │                                                                              │
 ╚─────────────────────────────────────────────────────────────────────────────*/
+#include "libc/log/log.h"
 #include "third_party/ggml/ggml.h"
 #include "third_party/libcxx/cstdio"
 #include "third_party/libcxx/map"
@@ -47,6 +48,8 @@ static const std::map<std::string, enum gptneox_ftype> GPTNEOX_FTYPE_MAP = {
 //  ./quantize models/llama/ggml-model.bin models/llama/ggml-model-quant.bin type
 //
 int main(int argc, char ** argv) {
+    ShowCrashReports();
+
     ggjt_v1();
     ggml_time_init();
 
diff --git a/third_party/radpajama/radpajama.mk b/third_party/radpajama/radpajama.mk
index 96107e46c..e4a103ef0 100644
--- a/third_party/radpajama/radpajama.mk
+++ b/third_party/radpajama/radpajama.mk
@@ -79,6 +79,7 @@ THIRD_PARTY_RADPAJAMA_MAIN_DIRECTDEPS =					\
 	LIBC_CALLS							\
 	LIBC_FMT							\
 	LIBC_INTRIN							\
+	LIBC_LOG							\
 	LIBC_NEXGEN32E							\
 	LIBC_RUNTIME							\
 	LIBC_STDIO							\