43 #include "Teuchos_UnitTestHarness.hpp" 44 #include "Teuchos_UnitTestRepository.hpp" 45 #include "Teuchos_GlobalMPISession.hpp" 48 #include "Kokkos_Core.hpp" 53 #ifdef HAVE_STOKHOS_MKL 73 #ifdef HAVE_STOKHOS_MKL 76 typedef Kokkos::OpenMP Device;
77 typedef Stokhos::MKLMultiply SparseMatOps;
78 success = test_crs_matrix_free<Scalar,Device,SparseMatOps>(
84 Teuchos::GlobalMPISession mpiSession(&argc, &
argv);
86 const size_t team_count =
87 Kokkos::hwloc::get_available_numa_count() *
88 Kokkos::hwloc::get_available_cores_per_numa();
89 const size_t threads_per_team =
90 Kokkos::hwloc::get_available_threads_per_core();
93 Kokkos::InitArguments init_args;
94 init_args.num_threads = team_count*threads_per_team;
95 Kokkos::initialize( init_args );
102 int ret = Teuchos::UnitTestRepository::runUnitTestsFromMain(argc,
argv);
UnitTestSetup< Kokkos::OpenMP > setup
int main(int argc, char *argv[])
#define UNIT_TEST_GROUP_SCALAR_HOST_DEVICE(SCALAR, DEVICE)
#define UNIT_TEST_GROUP_SCALAR_DEVICE(SCALAR, DEVICE)