29 #include "Teuchos_UnitTestHarness.hpp" 30 #include "Teuchos_UnitTestRepository.hpp" 31 #include "Teuchos_GlobalMPISession.hpp" 34 #define SACADO_VIEW_CUDA_HIERARCHICAL_DFAD 1 35 #define SACADO_KOKKOS_USE_MEMORY_POOL 1 41 #undef VIEW_FAD_TESTS_FDC 42 #define VIEW_FAD_TESTS_FDC( F, D ) \ 43 VIEW_FAD_TESTS_FLD( F, LeftContiguous32, D ) \ 44 VIEW_FAD_TESTS_FLD( F, RightContiguous32, D ) 46 #undef VIEW_FAD_TESTS_SFDC 47 #define VIEW_FAD_TESTS_SFDC( F, D ) \ 48 VIEW_FAD_TESTS_SFLD( F, LeftContiguous32, D ) \ 49 VIEW_FAD_TESTS_SFLD( F, RightContiguous32, D ) 52 #if defined(KOKKOS_ENABLE_CUDA_UVM) 57 int main(
int argc,
char* argv[] ) {
58 Teuchos::GlobalMPISession mpiSession(&argc, &argv);
61 Kokkos::InitArguments init_args;
62 init_args.device_id = 0;
63 Kokkos::initialize( init_args );
64 Kokkos::print_configuration(std::cout);
66 #if defined(SACADO_KOKKOS_USE_MEMORY_POOL) 75 int res = Teuchos::UnitTestRepository::runUnitTestsFromMain(argc, argv);
77 #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
Kokkos::LayoutContiguous< Kokkos::LayoutRight, 32 > RightContiguous32
Kokkos::LayoutContiguous< Kokkos::LayoutLeft, 32 > LeftContiguous32
const int global_num_cols
int main(int argc, char *argv[])
#define VIEW_FAD_TESTS_FDC(F, D)
void destroyGlobalMemoryPool(const ExecSpace &space)