Updating Kokkos lib
This commit is contained in:
@ -154,13 +154,13 @@ int main(int argc, char** argv) {
|
||||
|
||||
printf("-> Init Device\n");
|
||||
|
||||
#if defined( KOKKOS_HAVE_CUDA )
|
||||
#if defined( KOKKOS_ENABLE_CUDA )
|
||||
Kokkos::HostSpace::execution_space::initialize(teams*num_threads);
|
||||
Kokkos::Cuda::SelectDevice select_device(device);
|
||||
Kokkos::Cuda::initialize(select_device);
|
||||
#elif defined( KOKKOS_HAVE_OPENMP )
|
||||
#elif defined( KOKKOS_ENABLE_OPENMP )
|
||||
Kokkos::OpenMP::initialize(teams*num_threads);
|
||||
#elif defined( KOKKOS_HAVE_PTHREAD )
|
||||
#elif defined( KOKKOS_ENABLE_PTHREAD )
|
||||
Kokkos::Threads::initialize(teams*num_threads);
|
||||
#endif
|
||||
|
||||
|
||||
Reference in New Issue
Block a user