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):