Add a possible msvc fix

JamesYang007 · Oct 14, 2024 · 4c6b3c0 · 4c6b3c0
1 parent 86a160a
commit 4c6b3c0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/adelie/src/include/adelie_core/matrix/matrix_naive_convex_relu.hpp b/adelie/src/include/adelie_core/matrix/matrix_naive_convex_relu.hpp
@@ -184,14 +184,15 @@ class MatrixNaiveConvexReluDense: public MatrixNaiveBase<typename DenseType::Sca
     {
         const auto d = _mat.cols();
         const auto m = _mask.cols();
+        const auto v_weights = (v * weights).matrix();
+        Eigen::Map<rowmat_value_t> buff(_buff.data(), _n_threads, d);
         const auto routine = [&](auto i) {
             const auto i_sgn = i / m;
             const auto i_m = i - i_sgn * m;
             auto out_m = out.segment(i * d, d).matrix();
-            Eigen::Map<rowmat_value_t> buff(_buff.data(), _n_threads, d);
             dgemv(
                 _mat,
-                (1-2*i_sgn) * _mask.col(i_m).transpose().template cast<value_t>().cwiseProduct((v * weights).matrix()),
+                (1-2*i_sgn) * _mask.col(i_m).transpose().template cast<value_t>().cwiseProduct(v_weights),
                 1,
                 buff /* unused */,
                 out_m