From 7a12e29027b5d2ddce9238d6cd8331cfd4d76d95 Mon Sep 17 00:00:00 2001
From: Jan Wassenberg <janwas@google.com>
Date: Fri, 26 Apr 2024 10:59:05 -0700
Subject: [PATCH] Add error-checking for py binding, add missing include+hwasan
 check

PiperOrigin-RevId: 628453112
---
 BUILD.bazel | 1 +
 util/app.h  | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/BUILD.bazel b/BUILD.bazel
index 47e8175..bca3fd8 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -89,6 +89,7 @@ cc_library(
     deps = [
         ":args",
         ":gemma_lib",
+        "//compression:io",
         "@hwy//:hwy",
     ],
 )
diff --git a/util/app.h b/util/app.h
index 1a96c22..6541688 100644
--- a/util/app.h
+++ b/util/app.h
@@ -31,6 +31,7 @@
 #include <algorithm>  // std::clamp
 #include <thread>     // NOLINT>
 
+#include "compression/io.h"  // Path
 #include "gemma/configs.h"
 #include "gemma/gemma.h"
 #include "util/args.h"
@@ -45,6 +46,10 @@ static inline const char* CompiledConfig() {
     return "msan";
   } else if (HWY_IS_TSAN) {
     return "tsan";
+#if defined(HWY_IS_HWASAN)
+  } else if (HWY_IS_HWASAN) {
+    return "hwasan";
+#endif
 #if defined(HWY_IS_UBSAN)
   } else if (HWY_IS_UBSAN) {
     return "ubsan";