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 // std::clamp #include // 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";