1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
# Provides the ruy_test macro for type-parametrized tests.
"""ruy_test is a macro for building a test with multiple paths corresponding to tuples of types for LHS, RHS, accumulator and destination."""
def ruy_test(name, srcs, lhs_rhs_accum_dst, copts, tags = [], deps = None):
for (lhs, rhs, accum, dst) in lhs_rhs_accum_dst:
native.cc_test(
name = "%s_%s_%s_%s_%s" % (name, lhs, rhs, accum, dst),
srcs = srcs,
copts = copts + [
"-DRUY_TEST_LHSSCALAR=%s" % lhs,
"-DRUY_TEST_RHSSCALAR=%s" % rhs,
"-DRUY_TEST_ACCUMSCALAR=%s" % accum,
"-DRUY_TEST_DSTSCALAR=%s" % dst,
],
deps = deps,
tags = tags,
)
def ruy_benchmark(name, srcs, lhs_rhs_accum_dst, copts, deps = None):
tags = ["req_dep=//third_party/gemmlowp:profiler"]
for (lhs, rhs, accum, dst) in lhs_rhs_accum_dst:
native.cc_binary(
name = "%s_%s_%s_%s_%s" % (name, lhs, rhs, accum, dst),
testonly = True,
srcs = srcs,
copts = copts + [
"-DRUY_TEST_LHSSCALAR=%s" % lhs,
"-DRUY_TEST_RHSSCALAR=%s" % rhs,
"-DRUY_TEST_ACCUMSCALAR=%s" % accum,
"-DRUY_TEST_DSTSCALAR=%s" % dst,
],
deps = deps,
tags = tags,
)
|
