30 #include "Teuchos_UnitTestHarness.hpp" 31 #include "Teuchos_UnitTestRepository.hpp" 32 #include "Teuchos_GlobalMPISession.hpp" 35 #include "Kokkos_Core.hpp" 64 #ifdef KOKKOS_HAVE_OPENMP 65 #define TILED_CRS_TEST(SCALAR, DEVICE) 67 #define TILED_CRS_TEST(SCALAR, DEVICE) \ 68 TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( Kokkos_SG_SpMv, TiledCrsProductTensor, SCALAR, DEVICE ) 71 #define UNIT_TEST_GROUP_SCALAR_CUDA( SCALAR ) \ 72 TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( Kokkos_SG_SpMv, CrsMatrixFree, SCALAR, Cuda ) \ 73 TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( Kokkos_SG_SpMv, CrsMatrixFreeView, SCALAR, Cuda ) \ 74 TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( Kokkos_SG_SpMv, CrsMatrixFreeKokkos, SCALAR, Cuda ) \ 75 TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( Kokkos_SG_SpMv, CrsMatrixFreeSingleCol, SCALAR, Cuda ) \ 76 TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( Kokkos_SG_SpMv, CrsDenseBlock, SCALAR, Cuda ) \ 77 TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( Kokkos_SG_SpMv, CrsFlatCommuted, SCALAR, Cuda ) \ 78 TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( Kokkos_SG_SpMv, CrsFlatOriginal, SCALAR, Cuda ) \ 79 TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( Kokkos_SG_SpMv, CrsProductTensor, SCALAR, Cuda ) \ 80 TILED_CRS_TEST(SCALAR, Cuda ) \ 81 TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( Kokkos_SG_SpMv, CooProductTensorPacked, SCALAR, Cuda ) \ 82 TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( Kokkos_SG_SpMv, CooProductTensorUnpacked, SCALAR, Cuda ) \ 83 TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( Kokkos_SG_SpMv, LinearTensorSymmetric, SCALAR, Cuda ) \ 84 TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( Kokkos_SG_SpMv, LinearTensorAsymmetric, SCALAR, Cuda ) 92 Teuchos::GlobalMPISession mpiSession(&argc, &
argv);
95 Kokkos::HostSpace::execution_space::initialize();
96 Kokkos::Cuda::initialize( Kokkos::Cuda::SelectDevice(0) );
97 Kokkos::Cuda::print_configuration( std::cout );
103 int ret = Teuchos::UnitTestRepository::runUnitTestsFromMain(argc,
argv);
106 Kokkos::HostSpace::execution_space::finalize();
107 Kokkos::Cuda::finalize();
#define UNIT_TEST_GROUP_SCALAR_CUDA(SCALAR)
UnitTestSetup< Kokkos::Cuda > setup
int main(int argc, char *argv[])