summaryrefslogtreecommitdiffhomepage
path: root/test/benchmarks/ml/BUILD
diff options
context:
space:
mode:
authorZach Koopmans <zkoopmans@google.com>2020-12-07 16:17:14 -0800
committergVisor bot <gvisor-bot@google.com>2020-12-07 16:20:12 -0800
commit6d30688bd70370fd9327cd3fca178bb85342cc2a (patch)
tree6a806c2c5d9ae0a8f91180f73422df1d69b4a8b4 /test/benchmarks/ml/BUILD
parent615c3380d297ee3c61e47fb48a4cca87f7c99ba0 (diff)
Fix tags on benchmark targets.
PiperOrigin-RevId: 346203209
Diffstat (limited to 'test/benchmarks/ml/BUILD')
-rw-r--r--test/benchmarks/ml/BUILD5
1 files changed, 3 insertions, 2 deletions
diff --git a/test/benchmarks/ml/BUILD b/test/benchmarks/ml/BUILD
index 970f52706..02ff6966f 100644
--- a/test/benchmarks/ml/BUILD
+++ b/test/benchmarks/ml/BUILD
@@ -1,4 +1,5 @@
-load("//tools:defs.bzl", "go_library", "go_test")
+load("//tools:defs.bzl", "go_library")
+load("//test/benchmarks:defs.bzl", "benchmark_test")
package(licenses = ["notice"])
@@ -9,7 +10,7 @@ go_library(
deps = ["//test/benchmarks/harness"],
)
-go_test(
+benchmark_test(
name = "ml_test",
size = "large",
srcs = ["tensorflow_test.go"],