From 619ec982309cda5238d0ce7dc4a88bbb56eccbe5 Mon Sep 17 00:00:00 2001
From: Mengwei Liu <larryliu@fb.com>
Date: Wed, 7 Aug 2024 15:35:38 -0700
Subject: [PATCH] [llava][17/N] Move util.h into /e/llm/runner

So that it can be reused

[ghstack-poisoned]
---
 examples/models/llama2/runner/runner.cpp                | 2 +-
 examples/models/llama2/runner/targets.bzl               | 1 -
 extension/llm/runner/targets.bzl                        | 5 ++++-
 {examples/models/llama2 => extension/llm}/runner/util.h | 0
 4 files changed, 5 insertions(+), 3 deletions(-)
 rename {examples/models/llama2 => extension/llm}/runner/util.h (100%)

diff --git a/examples/models/llama2/runner/runner.cpp b/examples/models/llama2/runner/runner.cpp
index 2c8e48390ba..7e9d90cfe4e 100644
--- a/examples/models/llama2/runner/runner.cpp
+++ b/examples/models/llama2/runner/runner.cpp
@@ -27,7 +27,7 @@
 #include <torch/torch.h>
 #endif
 
-#include <executorch/examples/models/llama2/runner/util.h>
+#include <executorch/extension/llm/runner/util.h>
 #include <executorch/runtime/core/exec_aten/exec_aten.h>
 #include <executorch/runtime/core/exec_aten/util/scalar_type_util.h>
 #include <executorch/runtime/platform/log.h>
diff --git a/examples/models/llama2/runner/targets.bzl b/examples/models/llama2/runner/targets.bzl
index e67c9abd910..9800430b1f8 100644
--- a/examples/models/llama2/runner/targets.bzl
+++ b/examples/models/llama2/runner/targets.bzl
@@ -22,7 +22,6 @@ def define_common_targets():
             ],
             exported_headers = [
                 "runner.h",
-                "util.h",
             ],
             preprocessor_flags = [
                 "-DUSE_ATEN_LIB",
diff --git a/extension/llm/runner/targets.bzl b/extension/llm/runner/targets.bzl
index e0522c1ee57..2e37547437c 100644
--- a/extension/llm/runner/targets.bzl
+++ b/extension/llm/runner/targets.bzl
@@ -3,7 +3,10 @@ load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
 def define_common_targets():
     runtime.cxx_library(
         name = "stats",
-        exported_headers = ["stats.h"],
+        exported_headers = [
+            "stats.h",
+            "util.h",
+        ],
         visibility = [
             "@EXECUTORCH_CLIENTS",
         ],
diff --git a/examples/models/llama2/runner/util.h b/extension/llm/runner/util.h
similarity index 100%
rename from examples/models/llama2/runner/util.h
rename to extension/llm/runner/util.h