Update Kokkos library in LAMMPS to v3.5.0
This commit is contained in:
@ -47,7 +47,8 @@
|
||||
|
||||
#include <Kokkos_Macros.hpp>
|
||||
|
||||
#if defined(KOKKOS_ENABLE_MPI)
|
||||
//#define USE_MPI
|
||||
#if defined(USE_MPI)
|
||||
#include <mpi.h>
|
||||
#endif
|
||||
|
||||
@ -61,7 +62,7 @@ int main(int argc, char** argv) {
|
||||
|
||||
(void)argc;
|
||||
(void)argv;
|
||||
#if defined(KOKKOS_ENABLE_MPI)
|
||||
#if defined(USE_MPI)
|
||||
|
||||
MPI_Init(&argc, &argv);
|
||||
|
||||
@ -72,7 +73,7 @@ int main(int argc, char** argv) {
|
||||
msg << "MPI rank(" << mpi_rank << ") ";
|
||||
|
||||
#endif
|
||||
|
||||
Kokkos::initialize(argc, argv);
|
||||
msg << "{" << std::endl;
|
||||
|
||||
if (Kokkos::hwloc::available()) {
|
||||
@ -82,15 +83,13 @@ int main(int argc, char** argv) {
|
||||
<< std::endl;
|
||||
}
|
||||
|
||||
#if defined(KOKKOS_ENABLE_CUDA)
|
||||
Kokkos::Cuda::print_configuration(msg);
|
||||
#endif
|
||||
Kokkos::print_configuration(msg);
|
||||
|
||||
msg << "}" << std::endl;
|
||||
|
||||
std::cout << msg.str();
|
||||
|
||||
#if defined(KOKKOS_ENABLE_MPI)
|
||||
Kokkos::finalize();
|
||||
#if defined(USE_MPI)
|
||||
|
||||
MPI_Finalize();
|
||||
|
||||
|
||||
Reference in New Issue
Block a user