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::OpenMP::initialize( team_count * threads_per_team );
100 int ret = Teuchos::UnitTestRepository::runUnitTestsFromMain(argc,
argv);
103 Kokkos::OpenMP::finalize();
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)