Sacado Package Browser (Single Doxygen Collection)  Version of the Day
Fad_KokkosAtomicTests_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
29 #include "Teuchos_UnitTestHarness.hpp"
30 #include "Teuchos_UnitTestRepository.hpp"
31 #include "Teuchos_GlobalMPISession.hpp"
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 "Kokkos_Macros.hpp"
39 
40 #if defined(KOKKOS_ENABLE_CUDA_UVM)
41 #define SACADO_TEST_DFAD 1
42 #else
43 #define SACADO_TEST_DFAD 0
44 #endif
46 
49 #undef VIEW_FAD_TESTS_FD
50 #define VIEW_FAD_TESTS_FD( F, D ) \
51  VIEW_FAD_TESTS_FLD( F, LeftContiguous32, D ) \
52  VIEW_FAD_TESTS_FLD( F, RightContiguous32, D )
53 
54 // Instantiate tests for Cuda device
55 using Kokkos::Cuda;
56 VIEW_FAD_TESTS_D( Cuda )
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  );
75 #endif
76 
77  int res = Teuchos::UnitTestRepository::runUnitTestsFromMain(argc, argv);
78 
79 #if defined(SACADO_KOKKOS_USE_MEMORY_POOL)
80  Sacado::destroyGlobalMemoryPool(Kokkos::Cuda());
81 #endif
82 
83  // Finalize Cuda
84  Kokkos::finalize();
85 
86  return res;
87 }
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
Kokkos::LayoutContiguous< Kokkos::LayoutRight, 32 > RightContiguous32
const int global_num_rows
int main(int argc, char *argv[])
const int global_num_cols
void destroyGlobalMemoryPool(const ExecSpace &space)
#define VIEW_FAD_TESTS_D(D)
Kokkos::LayoutContiguous< Kokkos::LayoutLeft, 32 > LeftContiguous32