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 38 #include "Kokkos_Core.hpp" 42 #undef VIEW_FAD_TESTS_FDC 43 #define VIEW_FAD_TESTS_FDC( F, D ) \ 44 VIEW_FAD_TESTS_FLD( F, LeftContiguous32, D ) \ 45 VIEW_FAD_TESTS_FLD( F, RightContiguous32, D ) 47 #undef VIEW_FAD_TESTS_SFDC 48 #define VIEW_FAD_TESTS_SFDC( F, D ) \ 49 VIEW_FAD_TESTS_SFLD( F, LeftContiguous32, D ) \ 50 VIEW_FAD_TESTS_SFLD( F, RightContiguous32, D ) 53 #if defined(KOKKOS_ENABLE_CUDA_UVM) 58 int main(
int argc,
char* argv[] ) {
59 Teuchos::GlobalMPISession mpiSession(&argc, &argv);
62 Kokkos::InitArguments init_args;
63 init_args.device_id = 0;
64 Kokkos::initialize( init_args );
65 Kokkos::print_configuration(std::cout);
67 #if defined(SACADO_KOKKOS_USE_MEMORY_POOL) 76 int res = Teuchos::UnitTestRepository::runUnitTestsFromMain(argc, argv);
78 #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)