From 604ebadb896a430b70805cdddbedd8107e4285f8 Mon Sep 17 00:00:00 2001
From: Haoyu Zhang <haoyuzhang@google.com>
Date: Wed, 11 Mar 2020 14:12:47 -0700
Subject: [PATCH] Increase shard count for remote_execution_test and reenable
 tsan test.

PiperOrigin-RevId: 300404009
Change-Id: I6874fd4fcc07ae32212f5f73eab4631f0938d1c8
---
 tensorflow/python/eager/BUILD | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tensorflow/python/eager/BUILD b/tensorflow/python/eager/BUILD
index 0fcde009ed8..5696915dc80 100644
--- a/tensorflow/python/eager/BUILD
+++ b/tensorflow/python/eager/BUILD
@@ -873,10 +873,9 @@ cuda_py_test(
     srcs = ["remote_execution_test.py"],
     grpc_enabled = True,
     python_version = "PY3",
-    shard_count = 2,
+    shard_count = 8,
     tags = [
         "no_oss",  # This test launches local server
-        "notsan",  # TODO(b/151240519) Re-enable.
     ],
     deps = [
         "//tensorflow/python:array_ops",