Add a move assignment operator for CUDAWrappers::SparseMatrix

diff --git a/include/deal.II/lac/cuda_sparse_matrix.h b/include/deal.II/lac/cuda_sparse_matrix.h
index f8a1bb11e26587f4adbf4647c0d1ea42ef25d4b9..4599622de6d2a339ae0d0e5eb9987bbdb0ad2db3 100644 (file)
--- a/include/deal.II/lac/cuda_sparse_matrix.h
+++ b/include/deal.II/lac/cuda_sparse_matrix.h
@@ -103,6 +103,18 @@ namespace CUDAWrappers
      */
     ~SparseMatrix();
 
+    /**
+     * Move assignment operator.
+     */
+    SparseMatrix &
+    operator=(CUDAWrappers::SparseMatrix<Number> &&);
+
+    /**
+     * Copy assignment is deleted.
+     */
+    SparseMatrix &
+    operator=(const CUDAWrappers::SparseMatrix<Number> &) = delete;
+
     /**
      * Reinitialize the sparse matrix. The sparse matrix on the host is copied
      * to the device and the elementes are reordered according to the format

diff --git a/source/lac/cuda_sparse_matrix.cu b/source/lac/cuda_sparse_matrix.cu
index dfd9d7956bf72cbbbb37e0174dc7052bdfd21871..455af2aae4e5f725c60847cae8a36b6ec595e9b1 100644 (file)
--- a/source/lac/cuda_sparse_matrix.cu
+++ b/source/lac/cuda_sparse_matrix.cu
@@ -250,6 +250,32 @@ namespace CUDAWrappers
 
 
 
+  template <typename Number>
+  SparseMatrix<Number> &
+  SparseMatrix<Number>::operator=(SparseMatrix<Number> &&other)
+  {
+    cusparse_handle  = other.cusparse_handle;
+    nnz              = other.nnz;
+    n_rows           = other.n_rows;
+    n_cols           = other.n_cols;
+    val_dev          = other.val_dev;
+    column_index_dev = other.column_index_dev;
+    row_ptr_dev      = other.row_ptr_dev;
+    descr            = other.descr;
+
+    other.nnz              = 0;
+    other.n_rows           = 0;
+    other.n_cols           = 0;
+    other.val_dev          = nullptr;
+    other.column_index_dev = nullptr;
+    other.row_ptr_dev      = nullptr;
+    other.descr            = nullptr;
+
+    return *this;
+  }
+
+
+
   template <typename Number>
   void
   SparseMatrix<Number>::reinit(