add communicator size check to copy/move assignment

ginkgo-project · Apr 21, 2022 · cd61a1d · cd61a1d
1 parent d07bdf4
commit cd61a1d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/core/distributed/matrix.cpp b/core/distributed/matrix.cpp
@@ -357,6 +357,8 @@ Matrix<ValueType, LocalIndexType, GlobalIndexType>::operator=(
     const Matrix& other)
 {
     if (this != &other) {
+        GKO_ASSERT_EQ(other.get_communicator().size(),
+                      this->get_communicator().size());
         this->set_size(other.get_size());
         diag_mtx_->copy_from(other.diag_mtx_.get());
         offdiag_mtx_->copy_from(other.offdiag_mtx_.get());
@@ -380,6 +382,8 @@ Matrix<ValueType, LocalIndexType, GlobalIndexType>::operator=(
     Matrix&& other) noexcept
 {
     if (this != &other) {
+        GKO_ASSERT_EQ(other.get_communicator().size(),
+                      this->get_communicator().size());
         this->set_size(other.get_size());
         other.set_size({});
         diag_mtx_->move_from(other.diag_mtx_.get());