merge fix

lanl · Oct 4, 2024 · 04b7b51 · 04b7b51
2 parents ce443d4 + d5bf5c7
commit 04b7b51
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/include/macros.h b/src/include/macros.h
@@ -445,11 +445,11 @@ THREAD_ID \
 teamMember.team_rank()
 
 #define \
-FOR_FIRST(i, x1, fcn) \
+FOR_FIRST(i, x0, x1, fcn) \
 Kokkos::parallel_for( \
-                        Kokkos::TeamPolicy<>( x1, Kokkos::AUTO, 32 ), \
+                        Kokkos::TeamPolicy<>( x1-x0, Kokkos::AUTO, 32 ), \
                         KOKKOS_LAMBDA ( const Kokkos::TeamPolicy<>::member_type &teamMember ) \
-                        { const int i = TEAM_ID; fcn} )
+                        { const int i = TEAM_ID + x0; fcn} )
 
 #define \
 FOR_SECOND(j, y0, y1, fcn) \