From 5adba3f43587849113dc276ad29faaeb83ef63e3 Mon Sep 17 00:00:00 2001
From: Jeremy Lau <lauj@google.com>
Date: Wed, 4 Nov 2020 12:04:46 -0800
Subject: [PATCH] Increase size of stack_ops_test to medium. It previously
 timed out occasionally.

PiperOrigin-RevId: 340701018
Change-Id: I9b28510ee806088a0466a42d6cca00d19441ea57
---
 tensorflow/python/kernel_tests/array_ops/BUILD | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/tensorflow/python/kernel_tests/array_ops/BUILD b/tensorflow/python/kernel_tests/array_ops/BUILD
index bdcd4fdb478..c10c44e1ef4 100644
--- a/tensorflow/python/kernel_tests/array_ops/BUILD
+++ b/tensorflow/python/kernel_tests/array_ops/BUILD
@@ -85,11 +85,8 @@ cuda_py_test(
 
 cuda_py_test(
     name = "stack_op_test",
-    size = "small",
+    size = "medium",
     srcs = ["stack_op_test.py"],
-    xla_tags = [
-        "no_cuda_asan",  # times out
-    ],
     deps = [
         "//tensorflow/python:array_ops",
         "//tensorflow/python:client_testlib",