]> https://gitweb.dealii.org/ - dealii.git/commitdiff
Add missing instantiations for ditributed CUDA vectors
authorBruno Turcksin <bruno.turcksin@gmail.com>
Sun, 13 Jan 2019 03:02:13 +0000 (22:02 -0500)
committerBruno Turcksin <bruno.turcksin@gmail.com>
Sun, 13 Jan 2019 03:02:13 +0000 (22:02 -0500)
source/lac/vector_memory.cc

index 0b5e84c3ae45155fd1096dc50b4acf1af3961388..f15f32800e75473214b10af2fba3a1785b5b9f4e 100644 (file)
@@ -35,6 +35,16 @@ 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>>;
+#  ifdef DEAL_II_WITH_MPI
+template class VectorMemory<
+  LinearAlgebra::distributed::Vector<float, MemorySpace::CUDA>>;
+template class VectorMemory<
+  LinearAlgebra::distributed::Vector<double, MemorySpace::CUDA>>;
+template class GrowingVectorMemory<
+  LinearAlgebra::distributed::Vector<float, MemorySpace::CUDA>>;
+template class GrowingVectorMemory<
+  LinearAlgebra::distributed::Vector<double, MemorySpace::CUDA>>;
+#  endif
 #endif
 
 namespace internal
@@ -51,6 +61,14 @@ namespace internal
         float>>::release_unused_memory();
       dealii::GrowingVectorMemory<dealii::LinearAlgebra::CUDAWrappers::Vector<
         double>>::release_unused_memory();
+#  ifdef DEAL_II_WITH_MPI
+      dealii::GrowingVectorMemory<
+        dealii::LinearAlgebra::distributed::Vector<float, MemorySpace::CUDA>>::
+        release_unused_memory();
+      dealii::GrowingVectorMemory<
+        dealii::LinearAlgebra::distributed::Vector<double, MemorySpace::CUDA>>::
+        release_unused_memory();
+#  endif
 #endif
     }
   } // namespace GrowingVectorMemoryImplementation

In the beginning the Universe was created. This has made a lot of people very angry and has been widely regarded as a bad move.

Douglas Adams


Typeset in Trocchi and Trocchi Bold Sans Serif.