From: Daniel Arndt <arndtd@ornl.gov>
Date: Wed, 16 Aug 2023 20:59:32 +0000 (-0400)
Subject: Fix cuda_tensor_product_kernels.h
X-Git-Tag: relicensing~584^2
X-Git-Url: https://gitweb.dealii.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=refs%2Fpull%2F15888%2Fhead;p=dealii.git

Fix cuda_tensor_product_kernels.h
---

diff --git a/include/deal.II/matrix_free/cuda_tensor_product_kernels.h b/include/deal.II/matrix_free/cuda_tensor_product_kernels.h
index 74bb8be7c6..c26dd9ab51 100644
--- a/include/deal.II/matrix_free/cuda_tensor_product_kernels.h
+++ b/include/deal.II/matrix_free/cuda_tensor_product_kernels.h
@@ -151,19 +151,16 @@ namespace CUDAWrappers
       if (in_place)
         team_member.team_barrier();
 
-      Kokkos::parallel_for(Kokkos::TeamThreadRange(team_member, n_q_points),
-                           [&](const int i, const int j) {
-                             const int          q_point = i + j * n_q_points_1d;
-                             const unsigned int destination_idx =
-                               (direction == 0) ? (j + n_q_points_1d * i) :
-                                                  (i + n_q_points_1d * j);
-
-                             if (add)
-                               Kokkos::atomic_add(&out(destination_idx),
-                                                  t[q_point]);
-                             else
-                               out(destination_idx) = t[q_point];
-                           });
+      Kokkos::parallel_for(thread_policy, [&](const int i, const int j) {
+        const int          q_point = i + j * n_q_points_1d;
+        const unsigned int destination_idx =
+          (direction == 0) ? (j + n_q_points_1d * i) : (i + n_q_points_1d * j);
+
+        if (add)
+          Kokkos::atomic_add(&out(destination_idx), t[q_point]);
+        else
+          out(destination_idx) = t[q_point];
+      });
     }