diff --git a/extension/llm/custom_ops/targets.bzl b/extension/llm/custom_ops/targets.bzl
index 099266de1bb..ded25054acc 100644
--- a/extension/llm/custom_ops/targets.bzl
+++ b/extension/llm/custom_ops/targets.bzl
@@ -54,7 +54,10 @@ def define_common_targets():
         srcs = [
             "sdpa_with_kv_cache.py",
         ],
-        visibility = ["//executorch/..."],
+        visibility = [
+            "//executorch/...",
+            "@EXECUTORCH_CLIENTS",
+        ],
         deps = [
             "//caffe2:torch",
         ],