43 #include "Teuchos_UnitTestHarness.hpp" 44 #include "Teuchos_UnitTestRepository.hpp" 45 #include "Teuchos_GlobalMPISession.hpp" 48 #include "Kokkos_Core.hpp" 65 using Kokkos::Threads;
70 Teuchos::GlobalMPISession mpiSession(&argc, &
argv);
72 const size_t team_count =
73 Kokkos::hwloc::get_available_numa_count() *
74 Kokkos::hwloc::get_available_cores_per_numa();
75 const size_t threads_per_team =
76 Kokkos::hwloc::get_available_threads_per_core();
79 Kokkos::InitArguments init_args;
80 init_args.num_threads = team_count*threads_per_team;
81 init_args.device_id = 0;
82 Kokkos::initialize( init_args );
83 Kokkos::print_configuration( std::cout );
89 int ret = Teuchos::UnitTestRepository::runUnitTestsFromMain(argc,
argv);
UnitTestSetup< Kokkos::Threads > setup
int main(int argc, char *argv[])
#define UNIT_TEST_GROUP_SCALAR_HOST_DEVICE(SCALAR, DEVICE)
#define UNIT_TEST_GROUP_SCALAR_DEVICE(SCALAR, DEVICE)