From f49641432965a1e8f4288be401cb5e4799967b26 Mon Sep 17 00:00:00 2001 From: Daniel Arndt Date: Wed, 28 Dec 2022 22:43:31 +0100 Subject: [PATCH] wqAnnotate Kokkos::parallel_for --- include/deal.II/lac/la_parallel_vector.templates.h | 5 +++++ include/deal.II/lac/vector_operations_internal.h | 12 ++++++++++++ 2 files changed, 17 insertions(+) diff --git a/include/deal.II/lac/la_parallel_vector.templates.h b/include/deal.II/lac/la_parallel_vector.templates.h index f0060403e5..f5ec0ad5ff 100644 --- a/include/deal.II/lac/la_parallel_vector.templates.h +++ b/include/deal.II/lac/la_parallel_vector.templates.h @@ -388,6 +388,7 @@ namespace LinearAlgebra Kokkos::View indices("indices", n_elements); Kokkos::parallel_for( + "import_elements: fill indices", Kokkos::RangePolicy(host_exec, 0, n_elements), @@ -416,6 +417,7 @@ namespace LinearAlgebra // Set the values in tmp_vector Kokkos::parallel_for( + "import_elements: set values tmp_vector", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, n_elements), @@ -431,6 +433,7 @@ namespace LinearAlgebra const size_type tmp_n_elements = tmp_index_set.n_elements(); Kokkos::realloc(indices, tmp_n_elements); Kokkos::parallel_for( + "import_elements: copy local elements to val", Kokkos::RangePolicy(host_exec, 0, n_elements), @@ -446,6 +449,7 @@ namespace LinearAlgebra if (operation == VectorOperation::add) Kokkos::parallel_for( + "import_elements: add values", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, n_elements), @@ -454,6 +458,7 @@ namespace LinearAlgebra }); else Kokkos::parallel_for( + "import_elements: set values", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, n_elements), diff --git a/include/deal.II/lac/vector_operations_internal.h b/include/deal.II/lac/vector_operations_internal.h index 92300d124e..b1317471db 100644 --- a/include/deal.II/lac/vector_operations_internal.h +++ b/include/deal.II/lac/vector_operations_internal.h @@ -2177,6 +2177,7 @@ namespace internal auto exec = typename ::dealii::MemorySpace::Default::kokkos_space:: execution_space{}; Kokkos::parallel_for( + "add_vector", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, size), @@ -2197,6 +2198,7 @@ namespace internal auto exec = typename ::dealii::MemorySpace::Default::kokkos_space:: execution_space{}; Kokkos::parallel_for( + "subtract_vector", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, size), @@ -2216,6 +2218,7 @@ namespace internal auto exec = typename ::dealii::MemorySpace::Default::kokkos_space:: execution_space{}; Kokkos::parallel_for( + "add_factor", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, size), @@ -2237,6 +2240,7 @@ namespace internal auto exec = typename ::dealii::MemorySpace::Default::kokkos_space:: execution_space{}; Kokkos::parallel_for( + "add_av", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, size), @@ -2263,6 +2267,7 @@ namespace internal auto exec = typename ::dealii::MemorySpace::Default::kokkos_space:: execution_space{}; Kokkos::parallel_for( + "add_avpbw", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, size), @@ -2286,6 +2291,7 @@ namespace internal auto exec = typename ::dealii::MemorySpace::Default::kokkos_space:: execution_space{}; Kokkos::parallel_for( + "sadd_xv", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, size), @@ -2310,6 +2316,7 @@ namespace internal auto exec = typename ::dealii::MemorySpace::Default::kokkos_space:: execution_space{}; Kokkos::parallel_for( + "sadd_xav", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, size), @@ -2337,6 +2344,7 @@ namespace internal auto exec = typename ::dealii::MemorySpace::Default::kokkos_space:: execution_space{}; Kokkos::parallel_for( + "sadd_xavbw", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, size), @@ -2359,6 +2367,7 @@ namespace internal auto exec = typename ::dealii::MemorySpace::Default::kokkos_space:: execution_space{}; Kokkos::parallel_for( + "multiply_factor", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, size), @@ -2379,6 +2388,7 @@ namespace internal auto exec = typename ::dealii::MemorySpace::Default::kokkos_space:: execution_space{}; Kokkos::parallel_for( + "scale", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, size), @@ -2400,6 +2410,7 @@ namespace internal auto exec = typename ::dealii::MemorySpace::Default::kokkos_space:: execution_space{}; Kokkos::parallel_for( + "equ_au", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, size), @@ -2426,6 +2437,7 @@ namespace internal auto exec = typename ::dealii::MemorySpace::Default::kokkos_space:: execution_space{}; Kokkos::parallel_for( + "equ_aubv", Kokkos::RangePolicy< ::dealii::MemorySpace::Default::kokkos_space::execution_space>( exec, 0, size), -- 2.39.5