Fix size=0 for QR

Author: WardBrian

stan/math/prim/fun/qr.hpp

@@ -25,7 +25,10 @@ std::tuple<Eigen::Matrix<value_type_t<EigMat>, Eigen::Dynamic, Eigen::Dynamic>,
 qr(const EigMat& m) {
   using matrix_t
       = Eigen::Matrix<value_type_t<EigMat>, Eigen::Dynamic, Eigen::Dynamic>;
-  check_nonzero_size("qr", "m", m);
+  if (unlikely(m.size() == 0)) {
+    return std::make_tuple(matrix_t(0, 0), matrix_t(0, 0));
+  }
+
   Eigen::HouseholderQR<matrix_t> qr(m.rows(), m.cols());
   qr.compute(m);
   matrix_t Q = qr.householderQ();

stan/math/prim/fun/qr_Q.hpp

@@ -21,7 +21,10 @@ Eigen::Matrix<value_type_t<EigMat>, Eigen::Dynamic, Eigen::Dynamic> qr_Q(
     const EigMat& m) {
   using matrix_t
       = Eigen::Matrix<value_type_t<EigMat>, Eigen::Dynamic, Eigen::Dynamic>;
-  check_nonzero_size("qr_Q", "m", m);
+  if (unlikely(m.size() == 0)) {
+    return matrix_t(0, 0);
+  }
+  
   Eigen::HouseholderQR<matrix_t> qr(m.rows(), m.cols());
   qr.compute(m);
   matrix_t Q = qr.householderQ();

stan/math/prim/fun/qr_R.hpp

@@ -20,7 +20,9 @@ Eigen::Matrix<value_type_t<EigMat>, Eigen::Dynamic, Eigen::Dynamic> qr_R(
     const EigMat& m) {
   using matrix_t
       = Eigen::Matrix<value_type_t<EigMat>, Eigen::Dynamic, Eigen::Dynamic>;
-  check_nonzero_size("qr_R", "m", m);
+  if (unlikely(m.size() == 0)) {
+    return matrix_t(0, 0);
+  }
   Eigen::HouseholderQR<matrix_t> qr(m.rows(), m.cols());
   qr.compute(m);
   matrix_t R = qr.matrixQR();

stan/math/prim/fun/qr_thin.hpp

@@ -25,7 +25,10 @@ std::tuple<Eigen::Matrix<value_type_t<EigMat>, Eigen::Dynamic, Eigen::Dynamic>,
 qr_thin(const EigMat& m) {
   using matrix_t
       = Eigen::Matrix<value_type_t<EigMat>, Eigen::Dynamic, Eigen::Dynamic>;
-  check_nonzero_size("qr_thin", "m", m);
+  if (unlikely(m.size() == 0)) {
+    return std::make_tuple(matrix_t(0, 0), matrix_t(0, 0));
+  }
+
   Eigen::HouseholderQR<matrix_t> qr(m.rows(), m.cols());
   qr.compute(m);
   const int min_size = std::min(m.rows(), m.cols());

stan/math/prim/fun/qr_thin_Q.hpp

@@ -20,7 +20,9 @@ Eigen::Matrix<value_type_t<EigMat>, Eigen::Dynamic, Eigen::Dynamic> qr_thin_Q(
     const EigMat& m) {
   using matrix_t
       = Eigen::Matrix<value_type_t<EigMat>, Eigen::Dynamic, Eigen::Dynamic>;
-  check_nonzero_size("qr_thin_Q", "m", m);
+  if (unlikely(m.size() == 0)) {
+    return matrix_t(0, 0);
+  }
   Eigen::HouseholderQR<matrix_t> qr(m.rows(), m.cols());
   qr.compute(m);
   const int min_size = std::min(m.rows(), m.cols());

stan/math/prim/fun/qr_thin_R.hpp

@@ -20,7 +20,9 @@ Eigen::Matrix<value_type_t<EigMat>, Eigen::Dynamic, Eigen::Dynamic> qr_thin_R(
     const EigMat& m) {
   using matrix_t
       = Eigen::Matrix<value_type_t<EigMat>, Eigen::Dynamic, Eigen::Dynamic>;
-  check_nonzero_size("qr_thin_R", "m", m);
+  if (unlikely(m.size() == 0)) {
+    return matrix_t(0, 0);
+  }
   Eigen::HouseholderQR<matrix_t> qr(m.rows(), m.cols());
   qr.compute(m);
   const int min_size = std::min(m.rows(), m.cols());

test/unit/math/prim/fun/qr_Q_test.cpp

@@ -8,7 +8,7 @@ TEST(MathMatrixPrimMat, qr_Q) {
 
   using stan::math::qr_Q;
   using stan::math::transpose;
-  EXPECT_THROW(qr_Q(m0), std::invalid_argument);
+  EXPECT_NO_THROW(qr_Q(m0));
   EXPECT_NO_THROW(qr_Q(m1));
   EXPECT_NO_THROW(qr_Q(transpose(m1)));
 }

test/unit/math/prim/fun/qr_R_test.cpp

@@ -9,7 +9,7 @@ TEST(MathMatrixPrimMat, qr_R) {
   using stan::math::qr_Q;
   using stan::math::qr_R;
   using stan::math::transpose;
-  EXPECT_THROW(qr_R(m0), std::invalid_argument);
+  EXPECT_NO_THROW(qr_R(m0));
   EXPECT_NO_THROW(qr_R(m1));
 
   stan::math::matrix_d m2(4, 2);

test/unit/math/prim/fun/qr_test.cpp

@@ -8,7 +8,7 @@ TEST(MathMatrixPrimMat, qr) {
 
   using stan::math::qr;
   using stan::math::transpose;
-  EXPECT_THROW(qr(m0), std::invalid_argument);
+  EXPECT_NO_THROW(qr(m0));
   EXPECT_NO_THROW(qr(m1));
 
   stan::math::matrix_d m2(4, 2);

test/unit/math/prim/fun/qr_thin_Q_test.cpp

@@ -8,7 +8,7 @@ TEST(MathMatrixPrimMat, qr_thin_Q) {
 
   using stan::math::qr_thin_Q;
   using stan::math::transpose;
-  EXPECT_THROW(qr_thin_Q(m0), std::invalid_argument);
+  EXPECT_NO_THROW(qr_thin_Q(m0));
   EXPECT_NO_THROW(qr_thin_Q(m1));
   EXPECT_NO_THROW(qr_thin_Q(transpose(m1)));
 }