From 5477a9b314e7ae77d44044f904ad5a76a2c7e986 Mon Sep 17 00:00:00 2001 From: Stephen Bochinski Date: Fri, 19 Apr 2024 13:29:38 -0700 Subject: [PATCH] Expand visibility of targets needed for executorch_llama2 kernel Summary: See title Differential Revision: D56361946 --- examples/models/llama2/TARGETS | 2 ++ examples/models/llama2/tokenizer/targets.bzl | 1 + 2 files changed, 3 insertions(+) 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"],