Sacado Package Browser (Single Doxygen Collection)  Version of the Day
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Fad_KokkosTests_Cuda_Hierarchical_DFad.cpp
Go to the documentation of this file.
1 // @HEADER
2 // ***********************************************************************
3 //
4 // Sacado Package
5 // Copyright (2006) Sandia Corporation
6 //
7 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
8 // the U.S. Government retains certain rights in this software.
9 //
10 // This library is free software; you can redistribute it and/or modify
11 // it under the terms of the GNU Lesser General Public License as
12 // published by the Free Software Foundation; either version 2.1 of the
13 // License, or (at your option) any later version.
14 //
15 // This library is distributed in the hope that it will be useful, but
16 // WITHOUT ANY WARRANTY; without even the implied warranty of
17 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 // Lesser General Public License for more details.
19 //
20 // You should have received a copy of the GNU Lesser General Public
21 // License along with this library; if not, write to the Free Software
22 // Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301
23 // USA
24 // Questions? Contact David M. Gay (dmgay@sandia.gov) or Eric T. Phipps
25 // (etphipp@sandia.gov).
26 //
27 // ***********************************************************************
28 // @HEADER
32 
33 // Re-test cuda with hierarchical cuda parallelism turned on (experimental)
34 #define SACADO_VIEW_CUDA_HIERARCHICAL_DFAD 1
35 #define SACADO_KOKKOS_USE_MEMORY_POOL 1
36 
37 #include "Fad_KokkosTests.hpp"
38 #include "Kokkos_Core.hpp"
39 
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 )
46 
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 )
51 
52 // Instantiate tests for Cuda device
53 #if defined(KOKKOS_ENABLE_CUDA_UVM)
54 using Kokkos::Cuda;
56 #endif
57 
58 int main( int argc, char* argv[] ) {
59  Teuchos::GlobalMPISession mpiSession(&argc, &argv);
60 
61  // Initialize Cuda
62  Kokkos::InitArguments init_args;
63  init_args.device_id = 0;
64  Kokkos::initialize( init_args );
65  Kokkos::print_configuration(std::cout);
66 
67 #if defined(SACADO_KOKKOS_USE_MEMORY_POOL)
69  Kokkos::Cuda(),
71  global_fad_size*sizeof(double),
72  4*global_fad_size*sizeof(double),
73  128*global_fad_size*sizeof(double));
74 #endif
75 
77 
78 #if defined(SACADO_KOKKOS_USE_MEMORY_POOL)
79  Sacado::destroyGlobalMemoryPool(Kokkos::Cuda());
80 #endif
81 
82  // Finalize Cuda
83  Kokkos::finalize();
84 
85  return res;
86 }
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_num_rows
const int global_fad_size
static int runUnitTestsFromMain(int argc, char *argv[])
const int global_num_cols
int main()
Definition: ad_example.cpp:191
Kokkos::LayoutContiguous< Kokkos::LayoutLeft, 32 > LeftContiguous32
#define VIEW_FAD_TESTS_FDC(F, D)
void destroyGlobalMemoryPool(const ExecSpace &space)
Kokkos::LayoutContiguous< Kokkos::LayoutRight, 32 > RightContiguous32