| commit | 77344d076e523539be337587a8b8963452a53b72 | [log] [tgz] |
|---|---|---|
| author | Benoit Jacob <benoitjacob@google.com> | Thu Sep 19 16:04:25 2019 -0400 |
| committer | Benoit Jacob <benoitjacob@google.com> | Tue Mar 10 16:38:09 2020 -0400 |
| tree | a86f2ccffe7b9b7a61d32de568e2a329ad8a4dfc | |
| parent | fa906afaf5b23e4ab93e6414964e6fae2555360f [diff] |
Introduce pot_log2, which checks that its argument is a power of two then returns its log2. Allows to convey in code that values are power of two, so the log2 is exact. PiperOrigin-RevId: 270107047
ruy is a matrix multiplication library. Its focus is to cover the matrix multiplication needs of TensorFlow Lite.
ruy supports both floating-point (like Eigen) and quantized (like gemmlowp).
ruy is very new, immature code. It has quite good test coverage, but the code is in flux, lacks comments, needs more cleanup, and there are no design docs at the moment.
We hope to improve on all that and integrate ruy into TensorFlow Lite, at first as a non-default path for ARM A64 only, over the next few weeks [April 2019].
ruy is designed to achieve maximal performance not just on very large sizes, as is the focus of many established libraries, but on whatever are the actual sizes and shapes of matrices most critical in current TensorFlow Lite applications. This often means quite small sizes, e.g. 100x100 or even 50x50, and all sorts of rectangular shapes.
ruy is currently only optimized for ARM A64; other architectures have only slow reference code at the moment.
ruy is currently optimized only for the following combination of storage orders: LHS = row-major, RHS = column-major, destination = column-major. All other combinations of storage orders fall back to slow reference code at the moment.