From 6468a1d17ae43c161e836bd3d1af081e8e44ef0d Mon Sep 17 00:00:00 2001
From: Tres Popp <tpopp@google.com>
Date: Mon, 16 Nov 2020 01:44:55 -0800
Subject: [PATCH] Disable unranked kernels from tests.

PiperOrigin-RevId: 342588059
Change-Id: Icf48544ffc325e1b353389072f9f1d634d00de77
---
 tensorflow/core/kernels/mlir_generated/BUILD | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/tensorflow/core/kernels/mlir_generated/BUILD b/tensorflow/core/kernels/mlir_generated/BUILD
index b47fdbee1df..5f8c5ac95c4 100644
--- a/tensorflow/core/kernels/mlir_generated/BUILD
+++ b/tensorflow/core/kernels/mlir_generated/BUILD
@@ -82,7 +82,11 @@ cc_library(
 tf_kernel_library(
     name = "cwise_unary_op",
     srcs = [":unary_kernel_srcs"],
-    tags = ["manual"],
+    # TODO(herhut): Remove no_rocm once kernels build again.
+    tags = [
+        "manual",
+        "no_rocm",
+    ],
     deps = if_mlir_unranked_kernels_enabled(
         if_false = [
             ":abs_kernels",
@@ -115,7 +119,11 @@ tf_kernel_library(
 tf_kernel_library(
     name = "cwise_binary_op",
     srcs = ["unranked_gpu_add.cc"],
-    tags = ["manual"],
+    # TODO(herhut): Remove no_rocm once kernels build again.
+    tags = [
+        "manual",
+        "no_rocm",
+    ],
     deps = [
         ":addv2_unranked_kernels",
         ":unranked_op_gpu_base",
@@ -126,7 +134,11 @@ tf_kernel_library(
 tf_kernel_library(
     name = "cwise_op",
     srcs = [],
-    tags = ["no_rocm"],
+    # TODO(herhut): Remove manual once kernels build again.
+    tags = [
+        "manual",
+        "no_rocm",
+    ],
     # Technically these libraries don't need --config=cuda or --config=rocm,
     # but we want to avoid building them if they are not needed.
     deps = if_cuda_or_rocm([