int main( int argc, char* argv[] ) { Teuchos::GlobalMPISession mpiSession(&argc, &argv); setup.setup(); // Initialize host const std::pair<unsigned,unsigned> core_topo = KokkosArray::hwloc::get_core_topology(); //const size_t core_capacity = KokkosArray::hwloc::get_core_capacity(); const size_t gang_count = core_topo.first ; const size_t gang_worker_count = core_topo.second; KokkosArray::Host::initialize( gang_count , gang_worker_count ); #ifdef HAVE_KOKKOSARRAY_CUDA // Initialize Cuda KokkosArray::Cuda::initialize( KokkosArray::Cuda::SelectDevice(0) ); #endif // Run tests int ret = Teuchos::UnitTestRepository::runUnitTestsFromMain(argc, argv); // Finish up KokkosArray::Host::finalize(); #ifdef HAVE_KOKKOSARRAY_CUDA KokkosArray::Cuda::finalize(); #endif return ret; }
int main( int argc, char* argv[] ) { Teuchos::GlobalMPISession mpiSession(&argc, &argv); // Initialize threads Kokkos::Serial::initialize(); // Setup (has to happen after initialization) setup.setup(); // Run tests int ret = Teuchos::UnitTestRepository::runUnitTestsFromMain(argc, argv); // Finish up Kokkos::Serial::finalize(); return ret; }
TEUCHOS_UNIT_TEST_TEMPLATE_2_DECL( Kokkos_SG_SpMv, LinearTensorAsymmetric, Scalar, Device ) { const bool symmetric = false; UnitTestSetup<Device> s; s.setup(1, 10); success = test_linear_tensor<Scalar,Device,4>(s, out, symmetric); }