Skip to content

Commit bef8940

Browse files
committed
build: add portable kernel queue deps
1 parent 6e94c13 commit bef8940

1 file changed

Lines changed: 21 additions & 0 deletions

File tree

scripts/patch_tfjava.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1415,6 +1415,26 @@ def patch_workspace(path: Path) -> None:
14151415
"""
14161416

14171417

1418+
PORTABLE_KERNELS_QUEUE_DEPS_PATCH = """--- a/tensorflow/core/kernels/BUILD
1419+
+++ b/tensorflow/core/kernels/BUILD
1420+
@@ -6947,13 +6947,14 @@
1421+
],
1422+
visibility = ["//visibility:public"],
1423+
deps = [
1424+
"//tensorflow/core:portable_tensorflow_lib_lite",
1425+
"//tensorflow/core:protos_all_cc_impl",
1426+
+ ":padding_fifo_queue",
1427+
"//third_party/eigen3",
1428+
"//third_party/fft2d:fft2d_headers",
1429+
"@com_google_absl//absl/base",
1430+
"@com_google_protobuf//:protobuf",
1431+
"@fft2d",
1432+
"@gemmlowp",
1433+
],
1434+
alwayslink = 1,
1435+
"""
1436+
1437+
14181438
HUNK_HEADER_RE = re.compile(
14191439
r"^@@ -(?P<old_start>\d+)(?:,(?P<old_count>\d+))? "
14201440
r"\+(?P<new_start>\d+)(?:,(?P<new_count>\d+))? @@(?P<section>.*)$"
@@ -1501,6 +1521,7 @@ def write_tensorflow_android_absl_patch(path: Path) -> None:
15011521
text += SAVED_MODEL_ANDROID_LOADER_PATCH
15021522
text += TENSORFLOW_FRAMEWORK_ANDROID_PATCH
15031523
text += normalize_unified_diff_hunk_counts(ANDROID_PORTABLE_LIB_SHIM_PATCH)
1524+
text += PORTABLE_KERNELS_QUEUE_DEPS_PATCH
15041525
if not text.endswith("\n"):
15051526
text += "\n"
15061527
path.write_text(text, encoding="utf-8")

0 commit comments

Comments
 (0)