Skip to content

Commit 56f843b

Browse files
authored
Move metadata util to extension/llm/runner.
Differential Revision: D61108863 Pull Request resolved: #4664
1 parent 5e9bab8 commit 56f843b

File tree

5 files changed

+13
-2
lines changed

5 files changed

+13
-2
lines changed

examples/models/llama2/runner/runner.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
#include <executorch/extension/llm/tokenizer/bpe_tokenizer.h>
1717
#endif /* ET_USE_TIKTOKEN*/
1818
#include <executorch/extension/evalue_util/print_evalue.h>
19-
#include <executorch/extension/module/metadata_util.h>
19+
#include <executorch/extension/llm/runner/metadata_util.h>
2020
#include <executorch/extension/runner_util/managed_tensor.h>
2121

2222
#include <ctime>

examples/models/llama2/runner/targets.bzl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ def define_common_targets():
3232
],
3333
exported_deps = [
3434
"//executorch/backends/xnnpack:xnnpack_backend",
35+
"//executorch/extension/llm/runner:metadata_util" + aten_suffix,
3536
"//executorch/extension/llm/runner:stats",
3637
"//executorch/extension/llm/runner:text_decoder_runner" + aten_suffix,
3738
"//executorch/extension/llm/runner:text_prefiller" + aten_suffix,

extension/llm/runner/targets.bzl

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,3 +44,14 @@ def define_common_targets():
4444
"//executorch/extension/runner_util:managed_tensor" + aten_suffix,
4545
],
4646
)
47+
48+
runtime.cxx_library(
49+
name = "metadata_util" + aten_suffix,
50+
exported_headers = ["metadata_util.h"],
51+
visibility = [
52+
"@EXECUTORCH_CLIENTS",
53+
],
54+
exported_deps = [
55+
"//executorch/extension/module:module" + aten_suffix,
56+
],
57+
)

extension/module/targets.bzl

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@ def define_common_targets():
1717
],
1818
exported_headers = [
1919
"module.h",
20-
"metadata_util.h",
2120
],
2221
visibility = [
2322
"@EXECUTORCH_CLIENTS",

0 commit comments

Comments
 (0)