Ruy testing.

PiperOrigin-RevId: 247722924
This commit is contained in:
Benoit Jacob 2019-05-10 21:44:20 -07:00 committed by TensorFlower Gardener
parent 645ce7583c
commit fea8de6c36
2 changed files with 3 additions and 1 deletions

View File

@ -380,6 +380,7 @@ ruy_test(
("u8", "u8", "i32", "i16"),
("i8", "i8", "i32", "i32"),
],
tags = ["slow"],
)
ruy_test(

View File

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