From e3fb1f0fe6faf0be7fe2138e25e8f405a1adf1b9 Mon Sep 17 00:00:00 2001
From: stevebronder <stevo15025@gmail.com>
Date: Fri, 7 May 2021 13:35:55 -0400
Subject: [PATCH 1/2] csr_matrix_times_vector2 now uses csr_matrix_times_vector

---
 inst/include/csr_matrix_times_vector2.hpp | 22 ++++++++--------------
 1 file changed, 8 insertions(+), 14 deletions(-)

diff --git a/inst/include/csr_matrix_times_vector2.hpp b/inst/include/csr_matrix_times_vector2.hpp
index 9c708d10e..edbbec644 100644
--- a/inst/include/csr_matrix_times_vector2.hpp
+++ b/inst/include/csr_matrix_times_vector2.hpp
@@ -4,20 +4,14 @@
 /*
  * This works exactly like csr_matrix_times_vector but faster and less safe
  */
-template <typename T2__, typename T5__>
-inline
-Eigen::Matrix<typename boost::math::tools::promote_args<T2__, T5__>::type,
-              Eigen::Dynamic, 1>
-csr_matrix_times_vector2(const int& m,
-                         const int& n,
-                         const Eigen::Matrix<T2__, Eigen::Dynamic, 1>& w,
-                         const std::vector<int>& v,
-                         const std::vector<int>& u,
-                         const Eigen::Matrix<T5__, Eigen::Dynamic, 1>& b,
-                         std::ostream* pstream__) {
-  Eigen::Map<const Eigen::SparseMatrix<T2__,Eigen::RowMajor> >
-    sm(m, n, w.size(), &u[0], &v[0], &w[0]);
-  return sm * b;
+ template <typename T2__, typename T5__>
+ Eigen::Matrix<stan::promote_args_t<stan::value_type_t<T2__>,
+ stan::value_type_t<T5__>>, -1, 1>
+ csr_matrix_times_vector2(const int& m, const int& n, const T2__& w_arg__,
+                          const std::vector<int>& v,
+                          const std::vector<int>& u, const T5__& b_arg__,
+                          std::ostream* pstream__) {
+  return stan::math::csr_matrix_times_vector(m, n, w_arg__, v, u, b_arg__);
 }
 
 /* This specialization is slower than the above templated version

From e0fb4fcbfbf1f06d9a51829d69a84f9144831c67 Mon Sep 17 00:00:00 2001
From: stevebronder <stevo15025@gmail.com>
Date: Fri, 7 May 2021 20:24:35 -0400
Subject: [PATCH 2/2] also include the old version for backwards compatability

---
 inst/include/csr_matrix_times_vector2.hpp | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/inst/include/csr_matrix_times_vector2.hpp b/inst/include/csr_matrix_times_vector2.hpp
index edbbec644..19678a657 100644
--- a/inst/include/csr_matrix_times_vector2.hpp
+++ b/inst/include/csr_matrix_times_vector2.hpp
@@ -1,5 +1,21 @@
-#ifndef RSTANARM__CSR_MATRIX_TIMES_VECTOR2_HPP
-#define RSTANARM__CSR_MATRIX_TIMES_VECTOR2_HPP
+#ifndef RSTANARM_CSR_MATRIX_TIMES_VECTOR2_HPP
+#define RSTANARM_CSR_MATRIX_TIMES_VECTOR2_HPP
+
+template <typename T2__, typename T5__>
+inline
+Eigen::Matrix<typename boost::math::tools::promote_args<T2__, T5__>::type,
+              Eigen::Dynamic, 1>
+csr_matrix_times_vector2(const int& m,
+                         const int& n,
+                         const Eigen::Matrix<T2__, Eigen::Dynamic, 1>& w,
+                         const std::vector<int>& v,
+                         const std::vector<int>& u,
+                         const Eigen::Matrix<T5__, Eigen::Dynamic, 1>& b,
+                         std::ostream* pstream__) {
+  Eigen::Map<const Eigen::SparseMatrix<T2__,Eigen::RowMajor> >
+    sm(m, n, w.size(), &u[0], &v[0], &w[0]);
+  return sm * b;
+}
 
 /*
  * This works exactly like csr_matrix_times_vector but faster and less safe