ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Kokkos library in LAMMPS to v2.9.00

Browse Source
This commit is contained in:
Stan Moore
2019-06-28 11:23:24 -06:00
parent 7f342b1cd0
commit ea2e73119d
506 changed files with 37043 additions and 6851 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
lib/kokkos/example/multi_fem/TestCuda.cpp
View File

@ -86,12 +86,12 @@ void test_cuda_fixture( comm::Machine machine ,
dev_count && dev_count <= comm_size ? comm_rank % dev_count : 0 ;
const size_t gang_count = 0 ;
Kokkos::HostSpace::execution_space::initialize();
Kokkos::Cuda::SelectDevice select_device( dev_rank );
Kokkos::Cuda::initialize( select_device );
Kokkos::InitArguments args;
args.device_id = dev_rank;
Kokkos::initialize(args);
test_box_fixture<Kokkos::Cuda>( machine , gang_count , nx , ny , nz );
Kokkos::Cuda::finalize();
Kokkos::HostSpace::execution_space::finalize();
Kokkos::finalize();
}
//----------------------------------------------------------------------------
@ -108,12 +108,11 @@ void test_cuda_implicit( comm::Machine machine ,
dev_count && dev_count <= comm_size ? comm_rank % dev_count : 0 ;
const size_t gang_count = 0 ;
Kokkos::HostSpace::execution_space::initialize();
Kokkos::Cuda::SelectDevice select_device( dev_rank );
Kokkos::Cuda::initialize( select_device );
Kokkos::InitArguments args;
args.device_id = dev_rank;
Kokkos::initialize(args);
HybridFEM::Implicit::driver<double,Kokkos::Cuda>( "Cuda" , machine , gang_count , elem_count_begin , elem_count_end , count_run );
Kokkos::Cuda::finalize();
Kokkos::HostSpace::execution_space::finalize();
Kokkos::finalize();
}
//----------------------------------------------------------------------------
@ -130,12 +129,11 @@ void test_cuda_explicit( comm::Machine machine ,
dev_count && dev_count <= comm_size ? comm_rank % dev_count : 0 ;
const size_t gang_count = 0 ;
Kokkos::HostSpace::execution_space::initialize();
Kokkos::Cuda::SelectDevice select_device( dev_rank );
Kokkos::Cuda::initialize( select_device );
Kokkos::InitArguments args;
args.device_id = dev_rank;
Kokkos::initialize(args);
Explicit::driver<double,Kokkos::Cuda>( "Cuda" , machine , gang_count , elem_count_begin , elem_count_end , count_run );
Kokkos::Cuda::finalize();
Kokkos::HostSpace::execution_space::finalize();
Kokkos::finalize();
}
//----------------------------------------------------------------------------
@ -152,15 +150,14 @@ void test_cuda_nonlinear( comm::Machine machine ,
dev_count && dev_count <= comm_size ? comm_rank % dev_count : 0 ;
const size_t gang_count = 0 ;
Kokkos::HostSpace::execution_space::initialize();
Kokkos::Cuda::SelectDevice select_device( dev_rank );
Kokkos::Cuda::initialize( select_device );
Kokkos::InitArguments args;
args.device_id = dev_rank;
Kokkos::initialize(args);
typedef Kokkos::Cuda device ;
typedef FixtureElementHex8 hex8 ;
HybridFEM::Nonlinear::driver<double,device,hex8>( "Cuda" , machine , gang_count , elem_count_begin , elem_count_end , count_run );
Kokkos::Cuda::finalize();
Kokkos::HostSpace::execution_space::finalize();
Kokkos::finalize();
}
void test_cuda_nonlinear_quadratic( comm::Machine machine ,
@ -175,15 +172,14 @@ void test_cuda_nonlinear_quadratic( comm::Machine machine ,
dev_count && dev_count <= comm_size ? comm_rank % dev_count : 0 ;
const size_t gang_count = 0 ;
Kokkos::HostSpace::execution_space::initialize();
Kokkos::Cuda::SelectDevice select_device( dev_rank );
Kokkos::Cuda::initialize( select_device );
Kokkos::InitArguments args;
args.device_id = dev_rank;
Kokkos::initialize(args);
typedef Kokkos::Cuda device ;
typedef FixtureElementHex27 hex27 ;
HybridFEM::Nonlinear::driver<double,device,hex27>( "Cuda" , machine , gang_count , elem_count_begin , elem_count_end , count_run );
Kokkos::Cuda::finalize();
Kokkos::HostSpace::execution_space::finalize();
Kokkos::finalize();
}
//----------------------------------------------------------------------------