Fix logic error (caused by copy-pasting :/).

SC-SGS · Feb 16, 2025 · fb74d9e · fb74d9e
1 parent 3c7710a
commit fb74d9e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/include/plssvm/matrix.hpp b/include/plssvm/matrix.hpp
@@ -982,8 +982,8 @@ template <typename T, layout_type layout>
     using ssize_type = typename matrix<T, layout>::ssize_type;
     matrix<T, layout> res{ plssvm::shape{ lhs.num_rows(), rhs.num_cols() } };
 
-    const auto num_rows = static_cast<ssize_type>(lhs.num_rows());
-    const auto num_cols = static_cast<ssize_type>(lhs.num_cols());
+    const auto num_rows = static_cast<ssize_type>(res.num_rows());
+    const auto num_cols = static_cast<ssize_type>(res.num_cols());
 
 #pragma omp parallel for collapse(2) default(none) shared(lhs, rhs, res, num_rows, num_cols)
     for (ssize_type row = 0; row < num_rows; ++row) {