diff options
| author | Ahmed S. Taei <ataei@google.com> | 2020-09-16 18:11:17 +0000 |
|---|---|---|
| committer | Ahmed S. Taei <ataei@google.com> | 2020-09-22 04:53:57 +0000 |
| commit | 9b47525824df9be5ae23c39e7ce0d220d12f85e5 (patch) | |
| tree | 1f864ab16d196ee5d561c6f1da98444caf783f6c /clang/unittests/Tooling/Syntax/TreeTest.cpp | |
| parent | 9c7b58080ebd7ad3a63b892b5ae7f92e41587327 (diff) | |
| download | llvm-9b47525824df9be5ae23c39e7ce0d220d12f85e5.zip llvm-9b47525824df9be5ae23c39e7ce0d220d12f85e5.tar.gz llvm-9b47525824df9be5ae23c39e7ce0d220d12f85e5.tar.bz2 | |
Reorder linalg.conv indexing_maps loop order
Change the indexing map to iterate over the (b, x0, x1, z0, z1, q, k) instead of (b, x0, x1, k, q, z0, z1) to evaluate the convolution expression:
Y[b, x0, x1, k] = sum(W[z0, z1, q, k] * X[b, x0 + z0, x1 + z1, q], z0, z1, q)
This allows llvm auto vectorize to work and has better locality resulting significant performance improvments
Differential Revision: https://reviews.llvm.org/D87781
Diffstat (limited to 'clang/unittests/Tooling/Syntax/TreeTest.cpp')
0 files changed, 0 insertions, 0 deletions