diff options
| author | Alex Stark <starka@google.com> | 2019-08-21 17:17:19 +0300 |
|---|---|---|
| committer | Benoit Jacob <benoitjacob@google.com> | 2020-03-10 23:36:45 +0300 |
| commit | b7ebb1879201df49280c4fb449840abc3075287f (patch) | |
| tree | e8a2a7a38066123fac01e6499ba4e6d0cad53491 /BUILD | |
| parent | fa69a4bbdf3b676156668842b5d2e042cd4cd1f7 (diff) | |
Ruy: AVX2 model C++ code.
PiperOrigin-RevId: 264602338
Diffstat (limited to 'BUILD')
| -rw-r--r-- | BUILD | 34 |
1 files changed, 34 insertions, 0 deletions
@@ -356,6 +356,21 @@ cc_library( ) cc_library( + name = "kernel_avx2", + srcs = [ + "kernel_avx2.cc", + ], + copts = RUY_COPTS, + deps = [ + ":check_macros", + ":kernel_common", + ":opt_set", + ":platform", + "//third_party/gemmlowp:profiler", + ], +) + +cc_library( name = "kernel", hdrs = [ "kernel.h", @@ -367,6 +382,7 @@ cc_library( ":common", ":internal_matrix", ":kernel_arm", # fixdeps: keep + ":kernel_avx2", # fixdeps: keep ":kernel_avx512", # fixdeps: keep ":kernel_common", ":matrix", @@ -437,6 +453,23 @@ cc_library( ) cc_library( + name = "pack_avx2", + srcs = [ + "pack_avx2.cc", + ], + copts = RUY_COPTS, + deps = [ + ":check_macros", + ":matrix", + ":opt_set", + ":pack_common", + ":path", + ":platform", + "//third_party/gemmlowp:profiler", + ], +) + +cc_library( name = "pack", hdrs = [ "pack.h", @@ -450,6 +483,7 @@ cc_library( ":matrix", ":opt_set", ":pack_arm", # fixdeps: keep + ":pack_avx2", # fixdeps: keep ":pack_avx512", # fixdeps: keep ":pack_common", ":path", |