diff options
Diffstat (limited to 'test/benchmarks/ml')
| -rw-r--r-- | test/benchmarks/ml/BUILD | 5 |
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"], |