From 2aa15e7ae14691cd0cb6c8937228537f46d7e6bb Mon Sep 17 00:00:00 2001
From: Bruno Turcksin <bruno.turcksin@gmail.com>
Date: Fri, 13 Apr 2018 16:40:21 -0400
Subject: [PATCH] Instantiate GrowingVectorMemory for CUDAWrappers::Vector

---
 source/lac/vector_memory.cc | 13 +++++++++++++
 tests/cuda/solver_01.cu     |  3 ---
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/source/lac/vector_memory.cc b/source/lac/vector_memory.cc
index f94356da52..8a399fce73 100644
--- a/source/lac/vector_memory.cc
+++ b/source/lac/vector_memory.cc
@@ -15,6 +15,7 @@
 
 #include <deal.II/lac/vector.h>
 #include <deal.II/lac/block_vector.h>
+#include <deal.II/lac/cuda_vector.h>
 #include <deal.II/lac/la_vector.h>
 #include <deal.II/lac/la_parallel_vector.h>
 #include <deal.II/lac/la_parallel_block_vector.h>
@@ -30,6 +31,13 @@ DEAL_II_NAMESPACE_OPEN
 
 #include "vector_memory.inst"
 
+#ifdef DEAL_II_WITH_CUDA
+template class VectorMemory<LinearAlgebra::CUDAWrappers::Vector<float>>;
+template class VectorMemory<LinearAlgebra::CUDAWrappers::Vector<double>>;
+template class GrowingVectorMemory<LinearAlgebra::CUDAWrappers::Vector<float>>;
+template class GrowingVectorMemory<LinearAlgebra::CUDAWrappers::Vector<double>>;
+#endif
+
 namespace internal
 {
   namespace GrowingVectorMemoryImplementation
@@ -37,6 +45,11 @@ namespace internal
     void release_all_unused_memory()
     {
 #include "vector_memory_release.inst"
+
+#ifdef DEAL_II_WITH_CUDA
+      dealii::GrowingVectorMemory<dealii::LinearAlgebra::CUDAWrappers::Vector<float>>::release_unused_memory();
+      dealii::GrowingVectorMemory<dealii::LinearAlgebra::CUDAWrappers::Vector<double>>::release_unused_memory();
+#endif
     }
   }
 }
diff --git a/tests/cuda/solver_01.cu b/tests/cuda/solver_01.cu
index 1389a5be80..48940f0a62 100644
--- a/tests/cuda/solver_01.cu
+++ b/tests/cuda/solver_01.cu
@@ -18,9 +18,6 @@
 #include "../tests.h"
 #include "../testmatrix.h"
 
-#include <deal.II/lac/vector_memory.h>
-#include <deal.II/lac/vector_memory.templates.h>
-
 #include <deal.II/base/exceptions.h>
 #include <deal.II/lac/cuda_sparse_matrix.h>
 #include <deal.II/lac/precondition.h>
-- 
2.39.5