Increase memory limit for ASan builds of py_binary targets.

PiperOrigin-RevId: 908324533

Author: tensorflower-gardener

tensorflow_serving/experimental/example/BUILD

@@ -20,6 +20,10 @@ filegroup(
 py_binary(
     name = "remote_predict_client",
     srcs = ["remote_predict_client.py"],
+    exec_properties = select({
+        "//tools/cpp:asan_build": {"cpp_link.mem": "20g"},
+        "//conditions:default": None,
+    }),
     tags = [
         "manual",
     ],
@@ -31,6 +35,10 @@ py_binary(
     srcs = [
         "half_plus_two_with_rpop.py",
     ],
+    exec_properties = select({
+        "//tools/cpp:asan_build": {"cpp_link.mem": "20g"},
+        "//conditions:default": None,
+    }),
     srcs_version = "PY3",
     tags = [
         "manual",
@@ -41,6 +49,10 @@ py_binary(
 py_binary(
     name = "half_plus_two_with_rpop_client",
     srcs = ["half_plus_two_with_rpop_client.py"],
+    exec_properties = select({
+        "//tools/cpp:asan_build": {"cpp_link.mem": "20g"},
+        "//conditions:default": None,
+    }),
     tags = [
         "manual",
     ],

tensorflow_serving/model_servers/BUILD

@@ -641,6 +641,10 @@ py_binary(
 py_binary(
     name = "profiler_client",
     srcs = ["profiler_client.py"],
+    exec_properties = select({
+        "//tools/cpp:asan_build": {"cpp_link.mem": "20g"},
+        "//conditions:default": None,
+    }),
     srcs_version = "PY2AND3",
 )
 

tensorflow_serving/servables/tensorflow/testdata/BUILD

@@ -50,6 +50,10 @@ py_binary(
     srcs = [
         "export_bad_half_plus_two.py",
     ],
+    exec_properties = select({
+        "//tools/cpp:asan_build": {"cpp_link.mem": "20g"},
+        "//conditions:default": None,
+    }),
     srcs_version = "PY3",
 )
 
@@ -58,6 +62,10 @@ py_binary(
     srcs = [
         "export_counter.py",
     ],
+    exec_properties = select({
+        "//tools/cpp:asan_build": {"cpp_link.mem": "20g"},
+        "//conditions:default": None,
+    }),
     srcs_version = "PY3",
 )
 
@@ -66,6 +74,10 @@ py_binary(
     srcs = [
         "saved_model_half_plus_two.py",
     ],
+    exec_properties = select({
+        "//tools/cpp:asan_build": {"cpp_link.mem": "20g"},
+        "//conditions:default": None,
+    }),
     srcs_version = "PY3",
     deps = ["@org_tensorflow//tensorflow/lite/tools/signature:signature_def_utils"],
 )
@@ -75,6 +87,10 @@ py_binary(
     srcs = [
         "parse_example_tflite.py",
     ],
+    exec_properties = select({
+        "//tools/cpp:asan_build": {"cpp_link.mem": "20g"},
+        "//conditions:default": None,
+    }),
     main = "parse_example_tflite.py",
     srcs_version = "PY3",
     deps = ["@org_tensorflow//tensorflow/lite/tools/signature:signature_def_utils"],