Tpetra parallel linear algebra Version of the Day
Loading...
Searching...
No Matches
Tpetra_Details_initializeKokkos.cpp
1// @HEADER
2// ***********************************************************************
3//
4// Tpetra: Templated Linear Algebra Services Package
5// Copyright (2008) 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// Redistribution and use in source and binary forms, with or without
11// modification, are permitted provided that the following conditions are
12// met:
13//
14// 1. Redistributions of source code must retain the above copyright
15// notice, this list of conditions and the following disclaimer.
16//
17// 2. Redistributions in binary form must reproduce the above copyright
18// notice, this list of conditions and the following disclaimer in the
19// documentation and/or other materials provided with the distribution.
20//
21// 3. Neither the name of the Corporation nor the names of the
22// contributors may be used to endorse or promote products derived from
23// this software without specific prior written permission.
24//
25// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
26// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
29// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
30// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
31// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
32// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
33// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
34// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
35// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36//
37// Questions? Contact Michael A. Heroux (maherou@sandia.gov)
38//
39// ************************************************************************
40// @HEADER
41
43#include "Teuchos_GlobalMPISession.hpp"
44#include "Kokkos_Core.hpp"
45#include "Tpetra_Details_checkLaunchBlocking.hpp"
46#include <cstdlib> // std::atexit
47#include <string>
48#include <vector>
49
50namespace Tpetra {
51namespace Details {
52
53void finalizeKokkosIfNeeded() {
54 if(!Kokkos::is_finalized()) {
55 Kokkos::finalize();
56 }
57}
58
59void
61{
62 if (! Kokkos::is_initialized ()) {
63 std::vector<std::string> args = Teuchos::GlobalMPISession::getArgv ();
64 int narg = static_cast<int> (args.size ()); // must be nonconst
65
66 std::vector<char*> args_c;
67 std::vector<std::unique_ptr<char[]>> args_;
68 for (auto const& x : args) {
69 args_.emplace_back(new char[x.size() + 1]);
70 char* ptr = args_.back().get();
71 strcpy(ptr, x.c_str());
72 args_c.push_back(ptr);
73 }
74 args_c.push_back(nullptr);
75
76 Kokkos::initialize (narg, narg == 0 ? nullptr : args_c.data ());
77 checkOldCudaLaunchBlocking();
78
79 std::atexit (finalizeKokkosIfNeeded);
80 }
81}
82
83} // namespace Details
84} // namespace Tpetra
85
Declaration of Tpetra::Details::initializeKokkos.
Struct that holds views of the contents of a CrsMatrix.
Implementation details of Tpetra.
void initializeKokkos()
Initialize Kokkos, using command-line arguments (if any) given to Teuchos::GlobalMPISession.
Namespace Tpetra contains the class and methods constituting the Tpetra library.