Update benchmark to be a binary instead of test
PiperOrigin-RevId: 322477922 Change-Id: I1fd810f37e26e5141f96bbdd7dbb14bd84d8c1f6
This commit is contained in:
parent
4da55605f0
commit
147843127e
@ -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",
|
||||
|
Loading…
Reference in New Issue
Block a user