Fix Kokkos::parallel_for_pushforward signature for non-void work tag

gojakuch · Aug 27, 2024 · e2f4638 · e2f4638
1 parent dde2f08
commit e2f4638
Showing 1 changed file with 34 additions and 30 deletions.
diff --git a/include/clad/Differentiator/KokkosBuiltins.h b/include/clad/Differentiator/KokkosBuiltins.h
@@ -168,11 +168,11 @@ template <class Policy, class FunctorType, class T>
 struct diff_parallel_for_MDP_call_dispatch<Policy, FunctorType, T, 2> {
   static void run(const ::std::string& str, const Policy& policy,
                   const FunctorType& functor, const FunctorType& d_functor) {
-    ::Kokkos::parallel_for("_diff_" + str, policy,
-                           [&functor, &d_functor](const T x, auto&&... args) {
-                             functor.operator_call_pushforward(
-                                 x, args..., &d_functor, &x, 0, 0);
-                           });
+    ::Kokkos::parallel_for(
+        "_diff_" + str, policy,
+        [&functor, &d_functor](const auto x, auto&&... args) {
+          functor.operator_call_pushforward(x, args..., &d_functor, {}, 0, 0);
+        });
   }
 };
 template <class Policy, class FunctorType>
@@ -189,11 +189,12 @@ template <class Policy, class FunctorType, class T>
 struct diff_parallel_for_MDP_call_dispatch<Policy, FunctorType, T, 3> {
   static void run(const ::std::string& str, const Policy& policy,
                   const FunctorType& functor, const FunctorType& d_functor) {
-    ::Kokkos::parallel_for("_diff_" + str, policy,
-                           [&functor, &d_functor](const T x, auto&&... args) {
-                             functor.operator_call_pushforward(
-                                 x, args..., &d_functor, &x, 0, 0, 0);
-                           });
+    ::Kokkos::parallel_for(
+        "_diff_" + str, policy,
+        [&functor, &d_functor](const auto x, auto&&... args) {
+          functor.operator_call_pushforward(x, args..., &d_functor, {}, 0, 0,
+                                            0);
+        });
   }
 };
 template <class Policy, class FunctorType>
@@ -210,11 +211,12 @@ template <class Policy, class FunctorType, class T>
 struct diff_parallel_for_MDP_call_dispatch<Policy, FunctorType, T, 4> {
   static void run(const ::std::string& str, const Policy& policy,
                   const FunctorType& functor, const FunctorType& d_functor) {
-    ::Kokkos::parallel_for("_diff_" + str, policy,
-                           [&functor, &d_functor](const T x, auto&&... args) {
-                             functor.operator_call_pushforward(
-                                 x, args..., &d_functor, &x, 0, 0, 0, 0);
-                           });
+    ::Kokkos::parallel_for(
+        "_diff_" + str, policy,
+        [&functor, &d_functor](const auto x, auto&&... args) {
+          functor.operator_call_pushforward(x, args..., &d_functor, {}, 0, 0, 0,
+                                            0);
+        });
   }
 };
 template <class Policy, class FunctorType>
@@ -231,11 +233,12 @@ template <class Policy, class FunctorType, class T>
 struct diff_parallel_for_MDP_call_dispatch<Policy, FunctorType, T, 5> {
   static void run(const ::std::string& str, const Policy& policy,
                   const FunctorType& functor, const FunctorType& d_functor) {
-    ::Kokkos::parallel_for("_diff_" + str, policy,
-                           [&functor, &d_functor](const T x, auto&&... args) {
-                             functor.operator_call_pushforward(
-                                 x, args..., &d_functor, &x, 0, 0, 0, 0, 0);
-                           });
+    ::Kokkos::parallel_for(
+        "_diff_" + str, policy,
+        [&functor, &d_functor](const auto x, auto&&... args) {
+          functor.operator_call_pushforward(x, args..., &d_functor, {}, 0, 0, 0,
+                                            0, 0);
+        });
   }
 };
 template <class Policy, class FunctorType>
@@ -252,11 +255,12 @@ template <class Policy, class FunctorType, class T>
 struct diff_parallel_for_MDP_call_dispatch<Policy, FunctorType, T, 6> {
   static void run(const ::std::string& str, const Policy& policy,
                   const FunctorType& functor, const FunctorType& d_functor) {
-    ::Kokkos::parallel_for("_diff_" + str, policy,
-                           [&functor, &d_functor](const T x, auto&&... args) {
-                             functor.operator_call_pushforward(
-                                 x, args..., &d_functor, &x, 0, 0, 0, 0, 0, 0);
-                           });
+    ::Kokkos::parallel_for(
+        "_diff_" + str, policy,
+        [&functor, &d_functor](const auto x, auto&&... args) {
+          functor.operator_call_pushforward(x, args..., &d_functor, {}, 0, 0, 0,
+                                            0, 0, 0);
+        });
   }
 };
 template <class Policy, class FunctorType>
@@ -294,11 +298,11 @@ template <class Policy, class FunctorType, class T>
 struct diff_parallel_for_OP_call_dispatch {
   static void run(const ::std::string& str, const Policy& policy,
                   const FunctorType& functor, const FunctorType& d_functor) {
-    ::Kokkos::parallel_for("_diff_" + str, policy,
-                           [&functor, &d_functor](const T x, auto&&... args) {
-                             functor.operator_call_pushforward(
-                                 x, args..., &d_functor, &x, {});
-                           });
+    ::Kokkos::parallel_for(
+        "_diff_" + str, policy,
+        [&functor, &d_functor](const auto x, auto&&... args) {
+          functor.operator_call_pushforward(x, args..., &d_functor, {}, {});
+        });
   }
 };
 template <class Policy, class FunctorType>