From fea8de6c369fed68774eb5a263c0275720219910 Mon Sep 17 00:00:00 2001 From: Benoit Jacob Date: Fri, 10 May 2019 21:44:20 -0700 Subject: [PATCH] Ruy testing. PiperOrigin-RevId: 247722924 --- tensorflow/lite/experimental/ruy/BUILD | 1 + tensorflow/lite/experimental/ruy/ruy_test.bzl | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/tensorflow/lite/experimental/ruy/BUILD b/tensorflow/lite/experimental/ruy/BUILD index 85d7dda8e37..be1ddea2663 100644 --- a/tensorflow/lite/experimental/ruy/BUILD +++ b/tensorflow/lite/experimental/ruy/BUILD @@ -380,6 +380,7 @@ ruy_test( ("u8", "u8", "i32", "i16"), ("i8", "i8", "i32", "i32"), ], + tags = ["slow"], ) ruy_test( diff --git a/tensorflow/lite/experimental/ruy/ruy_test.bzl b/tensorflow/lite/experimental/ruy/ruy_test.bzl index c0a5000e879..df9f58ce653 100644 --- a/tensorflow/lite/experimental/ruy/ruy_test.bzl +++ b/tensorflow/lite/experimental/ruy/ruy_test.bzl @@ -6,7 +6,7 @@ corresponding to tuples of types for LHS, RHS, accumulator and destination. """ -def ruy_test(name, srcs, lhs_rhs_accum_dst): +def ruy_test(name, srcs, lhs_rhs_accum_dst, tags = []): for (lhs, rhs, accum, dst) in lhs_rhs_accum_dst: native.cc_test( name = "%s_%s_%s_%s_%s" % (name, lhs, rhs, accum, dst), @@ -21,6 +21,7 @@ def ruy_test(name, srcs, lhs_rhs_accum_dst): "//tensorflow/lite/experimental/ruy:test_lib", "@com_google_googletest//:gtest_main", ], + tags = tags, ) def ruy_benchmark(name, srcs, lhs_rhs_accum_dst):