From 147843127e8edb81cc2922bc6e1d16a43f6b4478 Mon Sep 17 00:00:00 2001
From: Akshay Modi <nareshmodi@google.com>
Date: Tue, 21 Jul 2020 17:34:25 -0700
Subject: [PATCH] Update benchmark to be a binary instead of test

PiperOrigin-RevId: 322477922
Change-Id: I1fd810f37e26e5141f96bbdd7dbb14bd84d8c1f6
---
 .../python/ops/numpy_ops/integration_test/benchmarks/BUILD   | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/tensorflow/python/ops/numpy_ops/integration_test/benchmarks/BUILD b/tensorflow/python/ops/numpy_ops/integration_test/benchmarks/BUILD
index 31f8d4575b3..012e1fbfcf3 100644
--- a/tensorflow/python/ops/numpy_ops/integration_test/benchmarks/BUILD
+++ b/tensorflow/python/ops/numpy_ops/integration_test/benchmarks/BUILD
@@ -1,5 +1,3 @@
-load("//tensorflow:tensorflow.bzl", "cuda_py_test")
-
 package(
     default_visibility = [
         "//tensorflow:internal",
@@ -7,11 +5,10 @@ package(
     licenses = ["notice"],  # Apache 2.0
 )
 
-cuda_py_test(
+py_binary(
     name = "micro_benchmarks",
     srcs = ["micro_benchmarks.py"],
     python_version = "PY3",
-    shard_count = 5,
     deps = [
         ":numpy_mlp",
         ":tf_numpy_mlp",