diff options
author | Benoit Jacob <benoitjacob@google.com> | 2019-07-25 20:58:42 +0300 |
---|---|---|
committer | Benoit Jacob <benoitjacob@google.com> | 2020-03-10 23:33:45 +0300 |
commit | a4aa186ebdf6ca1d88a49952440e495afb23a376 (patch) | |
tree | 71ce93ca2a0c57c17386a80ffa865554c8ccf617 /BUILD | |
parent | f9d85c737072a66dc417498b42de01504e1e67c3 (diff) |
Introduce a SidePair concept allowing us to rewrite much internal
ruy code taking advantage of LHS<->RHS code symmetry to remove
some redundancy.
The key motivation was that I want to experiment with some nontrivial
changes to how TrMulTask handles the packing of blocks, and I didn't want
to have to maintain two copies of this nontrivial code. With this change,
this code is now in a EnsurePacked method that's all I'll have to edit.
PiperOrigin-RevId: 259980220
Diffstat (limited to 'BUILD')
-rw-r--r-- | BUILD | 29 |
1 files changed, 27 insertions, 2 deletions
@@ -111,6 +111,12 @@ cc_test( ) cc_library( + name = "side_pair", + hdrs = ["side_pair.h"], + deps = [":check_macros"], +) + +cc_library( name = "block_map", srcs = [ "block_map.cc", @@ -121,6 +127,7 @@ cc_library( deps = [ ":check_macros", ":opt_set", + ":side_pair", ":size_util", "//third_party/gemmlowp:profiler", ], @@ -189,6 +196,7 @@ cc_library( ":block_map", ":check_macros", ":common", + ":side_pair", ":time", ], ) @@ -267,6 +275,7 @@ cc_library( ":opt_set", ":path", ":platform", + ":side_pair", ":size_util", ":spec", ":tune", @@ -296,20 +305,33 @@ cc_library( ) cc_library( + name = "trmul_params", + hdrs = ["trmul_params.h"], + deps = [ + ":context", + ":internal_matrix", + ":side_pair", + ":tune", + ], +) + +cc_library( name = "trmul", srcs = ["trmul.cc"], hdrs = ["trmul.h"], deps = [ ":allocator", ":block_map", + ":check_macros", ":common", ":context", ":internal_matrix", - ":kernel", ":opt_set", - ":pack", + ":side_pair", + ":spec", ":thread_pool", ":trace", + ":trmul_params", ":tune", "//third_party/gemmlowp:profiler", ], @@ -331,8 +353,11 @@ cc_library( ":check_macros", ":common", ":context", + ":kernel", ":matrix", + ":pack", ":path", + ":side_pair", ":size_util", ":spec", ":trmul", |