Fix size of data copied to constant memory and add missing include

diff --git a/include/deal.II/matrix_free/cuda_matrix_free.templates.h b/include/deal.II/matrix_free/cuda_matrix_free.templates.h
index 0d2fc60ad2106da32a92953818a1902314b7377b..1bd8d752854a6c13e64df11ea1f732c0af702c59 100644 (file)
--- a/include/deal.II/matrix_free/cuda_matrix_free.templates.h
+++ b/include/deal.II/matrix_free/cuda_matrix_free.templates.h
@@ -274,7 +274,9 @@ namespace CUDAWrappers
       cudaError_t error_code = cudaMemcpyToSymbol(
         constraint_weights,
         shape_info.data.front().subface_interpolation_matrix.data(),
-        sizeof(double) * fe.n_dofs_per_face(0) * fe.n_dofs_per_face(0));
+        sizeof(double) *
+          shape_info.data.front().subface_interpolation_matrix.size());
+      AssertCuda(error_code);
 
       local_dof_indices.resize(data->dofs_per_cell);
       lexicographic_dof_indices.resize(dofs_per_cell);

diff --git a/tests/cuda/cuda_vector_05.cu b/tests/cuda/cuda_vector_05.cu
index d11c5987747824d498f8d7d57b373d9d95a65e90..8c2dd8963d83e597aa2b418fb614f9998da99549 100644 (file)
--- a/tests/cuda/cuda_vector_05.cu
+++ b/tests/cuda/cuda_vector_05.cu
@@ -18,6 +18,7 @@
 
 #include <deal.II/lac/cuda_vector.h>
 #include <deal.II/lac/read_write_vector.h>
+#include <deal.II/lac/vector.h>
 
 #include <fstream>
 #include <iostream>