Skip to content

Instantly share code, notes, and snippets.

@DavidPoliakoff
Created July 24, 2020 14:01
Show Gist options
  • Save DavidPoliakoff/af22a41aa0575de36b7b82668e22651a to your computer and use it in GitHub Desktop.
Save DavidPoliakoff/af22a41aa0575de36b7b82668e22651a to your computer and use it in GitHub Desktop.
Kokkos::TeamPolicy<Kokkos::Cuda> execution_policy(total_amount_of_work,
outer_parallelism_batch_size,
inner_parallelism_batch_size);
Kokkos::parallel_for("implement_psychics", execution_policy,
KOKKOS_LAMBDA (team_member outer_handle) {
// some outer level work
double sum = psychics;
parallel_reduce (ThreadVectorRange (outer_handle, loop_count),
[=] (int& i, Scalar& lsum) {
// some inner level work
lsum += ...;
}, sum);
});
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment