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);
}
