FF (pytorch#21296)

Differential Revision: D113465092

Pull Request resolved: pytorch#21296

Author: metascroy

extension/llm/cache/BUCK

@@ -17,6 +17,19 @@ load(":targets.bzl", "define_common_targets")
 
 fbcode_target(_kind = define_common_targets,)
 
+fbcode_target(_kind = runtime.python_library,
+    name = "cache",
+    srcs = [
+        "reference_cache.py",
+        "update_and_attend.py",
+    ],
+    visibility = ["PUBLIC"],
+    deps = [
+        "//caffe2:torch",
+        "//executorch/exir:_warnings",
+    ],
+)
+
 fbcode_target(_kind = runtime.python_test,
     name = "test_update_and_attend",
     srcs = [

extension/llm/cache/targets.bzl

@@ -1,15 +1,2 @@
-load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
-
-
 def define_common_targets():
-    runtime.python_library(
-        name = "cache",
-        srcs = [
-            "reference_cache.py",
-            "update_and_attend.py",
-        ],
-        visibility = ["PUBLIC"],
-        deps = [
-            "//caffe2:torch",
-        ],
-    )
+    pass