Fix universal references (Issue #2881)

stan-dev · Mar 15, 2021 · 83f551f · 83f551f
1 parent 13d139f
commit 83f551f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/stan/mcmc/hmc/hamiltonians/dense_e_point.hpp b/src/stan/mcmc/hmc/hamiltonians/dense_e_point.hpp
@@ -37,8 +37,10 @@ class dense_e_point : public ps_point {
    *
    * @param inv_e_metric initial mass matrix
    */
-  void set_inv_metric(Eigen::MatrixXd&& inv_e_metric) {
-    inv_e_metric_ = std::forward<Eigen::MatrixXd>(inv_e_metric);
+  template <typename EigMat,
+	    require_eigen_matrix_dynamic_t<EigMat>* = nullptr>
+  void set_inv_metric(EigMat&& inv_e_metric) {
+    inv_e_metric_ = std::forward<EigMat>(inv_e_metric);
     inv_e_metric_llt_matrixU_ = inv_e_metric_.llt().matrixU();
   }
 

diff --git a/src/stan/mcmc/hmc/hamiltonians/diag_e_point.hpp b/src/stan/mcmc/hmc/hamiltonians/diag_e_point.hpp
@@ -34,8 +34,10 @@ class diag_e_point : public ps_point {
    *
    * @param inv_e_metric initial mass matrix
    */
-  void set_inv_metric(Eigen::VectorXd&& inv_e_metric) {
-    inv_e_metric_ = std::forward<Eigen::VectorXd>(inv_e_metric);
+  template <typename EigVec,
+	    require_eigen_vector_t<EigVec>* = nullptr>
+  void set_inv_metric(EigVec&& inv_e_metric) {
+    inv_e_metric_ = std::forward<EigVec>(inv_e_metric);
   }
 
   /**