29 #include "Teuchos_UnitTestHarness.hpp" 30 #include "Teuchos_UnitTestRepository.hpp" 31 #include "Teuchos_GlobalMPISession.hpp" 34 #define SACADO_VIEW_CUDA_HIERARCHICAL 1 35 #define SACADO_VIEW_CUDA_HIERARCHICAL_DFAD_STRIDED 1 36 #define SACADO_KOKKOS_USE_MEMORY_POOL 1 38 #include "Kokkos_Macros.hpp" 40 #if defined(KOKKOS_ENABLE_CUDA_UVM) 41 #define SACADO_TEST_DFAD 1 43 #define SACADO_TEST_DFAD 0 49 #undef VIEW_FAD_TESTS_FDC 50 #define VIEW_FAD_TESTS_FDC( F, D ) \ 51 VIEW_FAD_TESTS_FLD( F, LeftContiguous32, D ) \ 52 VIEW_FAD_TESTS_FLD( F, RightContiguous32, D ) 54 #undef VIEW_FAD_TESTS_SFDC 55 #define VIEW_FAD_TESTS_SFDC( F, D ) \ 56 VIEW_FAD_TESTS_SFLD( F, LeftContiguous32, D ) \ 57 VIEW_FAD_TESTS_SFLD( F, RightContiguous32, D ) 63 int
main(
int argc,
char* argv[] ) {
64 Teuchos::GlobalMPISession mpiSession(&argc, &argv);
67 Kokkos::InitArguments init_args;
68 init_args.device_id = 0;
69 Kokkos::initialize( init_args );
70 Kokkos::print_configuration(std::cout);
72 #if defined(SACADO_KOKKOS_USE_MEMORY_POOL) 82 int res = Teuchos::UnitTestRepository::runUnitTestsFromMain(argc, argv);
84 #if defined(SACADO_KOKKOS_USE_MEMORY_POOL) void createGlobalMemoryPool(const ExecSpace &space, const size_t min_total_alloc_size, const uint32_t min_block_alloc_size, const uint32_t max_block_alloc_size, const uint32_t min_superblock_size)
const int global_fad_size
const int global_num_rows
const int global_num_cols
int main(int argc, char *argv[])
Kokkos::LayoutContiguous< Kokkos::LayoutLeft, 32 > LeftContiguous32
void destroyGlobalMemoryPool(const ExecSpace &space)
#define VIEW_FAD_TESTS_D(D)
Kokkos::LayoutContiguous< Kokkos::LayoutRight, 32 > RightContiguous32