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::Threads::initialize( team_count * threads_per_team );
80 Kokkos::Threads::print_configuration( std::cout );
86 int ret = Teuchos::UnitTestRepository::runUnitTestsFromMain(argc,
argv);
89 Kokkos::Threads::finalize();
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)