From: Peter Munch <peterrmuench@gmail.com>
Date: Fri, 21 Oct 2022 07:29:57 +0000 (+0200)
Subject: TensorProductMatrix: with arbitrary number of lanes
X-Git-Tag: v9.5.0-rc1~885^2
X-Git-Url: https://gitweb.dealii.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=e68ca17c4cb6065818102c5cf9459d780ae07bfd;p=dealii.git

TensorProductMatrix: with arbitrary number of lanes
---

diff --git a/include/deal.II/lac/tensor_product_matrix.h b/include/deal.II/lac/tensor_product_matrix.h
index 86a32dcb26..e7d2fa09bb 100644
--- a/include/deal.II/lac/tensor_product_matrix.h
+++ b/include/deal.II/lac/tensor_product_matrix.h
@@ -585,16 +585,20 @@ namespace internal
 
 
 
-    template <std::size_t dim, typename Number>
+    template <std::size_t dim, typename Number, std::size_t n_lanes>
     inline void
-    setup(const std::array<Table<2, VectorizedArray<Number>>, dim> &mass_matrix,
-          const std::array<Table<2, VectorizedArray<Number>>, dim>
-            &                                                 derivative_matrix,
-          std::array<Table<2, VectorizedArray<Number>>, dim> &eigenvectors,
-          std::array<AlignedVector<VectorizedArray<Number>>, dim> &eigenvalues)
+    setup(
+      const std::array<Table<2, VectorizedArray<Number, n_lanes>>, dim>
+        &mass_matrix,
+      const std::array<Table<2, VectorizedArray<Number, n_lanes>>, dim>
+        &derivative_matrix,
+      std::array<Table<2, VectorizedArray<Number, n_lanes>>, dim> &eigenvectors,
+      std::array<AlignedVector<VectorizedArray<Number, n_lanes>>, dim>
+        &eigenvalues)
     {
-      const unsigned int     n_rows_1d   = mass_matrix[0].n_cols();
-      constexpr unsigned int macro_size  = VectorizedArray<Number>::size();
+      const unsigned int     n_rows_1d = mass_matrix[0].n_cols();
+      constexpr unsigned int macro_size =
+        VectorizedArray<Number, n_lanes>::size();
       const std::size_t nm_flat_size_max = n_rows_1d * n_rows_1d * macro_size;
       const std::size_t n_flat_size_max  = n_rows_1d * macro_size;