Skip to content

Commit de4ebf0

Browse files
Increase cpp_link.mem for ASAN builds to fix OOM failures.
PiperOrigin-RevId: 910953556
1 parent 792cd8a commit de4ebf0

2 files changed

Lines changed: 12 additions & 0 deletions

File tree

  • tensorflow_serving

tensorflow_serving/batching/test_util/BUILD

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,5 +50,11 @@ cc_test(
5050
py_binary(
5151
name = "matrix_half_plus_two_saved_model",
5252
srcs = ["matrix_half_plus_two_saved_model.py"],
53+
exec_properties = select({
54+
# copybara:comment_begin(google-only)
55+
"//tools/cpp:asan_build": {"cpp_link.mem": "20g"},
56+
# copybara:comment_end
57+
"//conditions:default": None,
58+
}),
5359
srcs_version = "PY3",
5460
)

tensorflow_serving/model_servers/BUILD

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -634,6 +634,12 @@ py_test(
634634
py_binary(
635635
name = "tensorflow_model_server_test_client",
636636
srcs = ["tensorflow_model_server_test_client.py"],
637+
exec_properties = select({
638+
# copybara:comment_begin(google-only)
639+
"//tools/cpp:asan_build": {"cpp_link.mem": "20g"},
640+
# copybara:comment_end
641+
"//conditions:default": None,
642+
}),
637643
srcs_version = "PY2AND3",
638644
deps = ["//tensorflow_serving/apis:prediction_service_proto_py_pb2"],
639645
)

0 commit comments

Comments
 (0)