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.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 1
35 #define SACADO_VIEW_CUDA_HIERARCHICAL_DFAD_STRIDED 1
36 #define SACADO_KOKKOS_USE_MEMORY_POOL 1
37 
38 #include "Fad_KokkosTests.hpp"
39 #include "Kokkos_Core.hpp"
40 
43 #undef VIEW_FAD_TESTS_FDC
44 #define VIEW_FAD_TESTS_FDC( F, D ) \
45  VIEW_FAD_TESTS_FLD( F, LeftContiguous32, D ) \
46  VIEW_FAD_TESTS_FLD( F, RightContiguous32, D )
47 
48 #undef VIEW_FAD_TESTS_SFDC
49 #define VIEW_FAD_TESTS_SFDC( F, D ) \
50  VIEW_FAD_TESTS_SFLD( F, LeftContiguous32, D ) \
51  VIEW_FAD_TESTS_SFLD( F, RightContiguous32, D )
52 
53 // Instantiate tests for Cuda device
54 #if defined(KOKKOS_ENABLE_CUDA_UVM)
55 #define SACADO_TEST_DFAD 1
56 #else
57 #define SACADO_TEST_DFAD 0
58 #endif
59 using Kokkos::Cuda;
60 VIEW_FAD_TESTS_D( Cuda )
61 
62 int main( int argc, char* argv[] ) {
63  Teuchos::GlobalMPISession mpiSession(&argc, &argv);
64 
65  // Initialize Cuda
66  Kokkos::InitArguments init_args;
67  init_args.device_id = 0;
68  Kokkos::initialize( init_args );
69  Kokkos::print_configuration(std::cout);
70 
71 #if defined(SACADO_KOKKOS_USE_MEMORY_POOL)
73  Kokkos::Cuda(),
75  global_fad_size*sizeof(double),
76  4*global_fad_size*sizeof(double),
77  128*global_fad_size*sizeof(double)
78  );
79 #endif
80 
82 
83 #if defined(SACADO_KOKKOS_USE_MEMORY_POOL)
84  Sacado::destroyGlobalMemoryPool(Kokkos::Cuda());
85 #endif
86 
87  // Finalize Cuda
88  Kokkos::finalize();
89 
90  return res;
91 }
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
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
void destroyGlobalMemoryPool(const ExecSpace &space)
#define VIEW_FAD_TESTS_D(D)
Kokkos::LayoutContiguous< Kokkos::LayoutRight, 32 > RightContiguous32