diff --git a/examples/models/llama2/TARGETS b/examples/models/llama2/TARGETS index 96c9e8b6ffd..d4d319b1d3a 100644 --- a/examples/models/llama2/TARGETS +++ b/examples/models/llama2/TARGETS @@ -36,6 +36,7 @@ runtime.python_library( }, visibility = [ "//bento/...", + "//bento_kernels/...", "//executorch/...", ], deps = [ @@ -74,6 +75,7 @@ runtime.python_library( base_module = "executorch.examples.models.llama2", visibility = [ "//bento/...", + "//bento_kernels/...", "//executorch/examples/...", ], deps = [ diff --git a/examples/models/llama2/tokenizer/targets.bzl b/examples/models/llama2/tokenizer/targets.bzl index 9c3d87a7728..51fdad8fa06 100644 --- a/examples/models/llama2/tokenizer/targets.bzl +++ b/examples/models/llama2/tokenizer/targets.bzl @@ -51,6 +51,7 @@ def define_common_targets(): visibility = [ "//executorch/examples/...", "//bento/...", + "//bento_kernels/...", ], _is_external_target = True, deps = [] if runtime.is_oss else ["fbsource//third-party/pypi/sentencepiece:sentencepiece"],