Tpetra_Details_ExecutionSpacesSlot.hpp

// @HEADER
// *****************************************************************************
//          Tpetra: Templated Linear Algebra Services Package
//
// Copyright 2008 NTESS and the Tpetra contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef TPETRA_DETAILS_EXEUTIONSPACESSLOT_HPP
#define TPETRA_DETAILS_EXEUTIONSPACESSLOT_HPP

#include <iostream>
#include <sstream>
#include <vector>

#include "Teuchos_RCP.hpp"
#include "Tpetra_Details_Behavior.hpp"
#include "Tpetra_Details_ExecutionSpaces.hpp"
#include <Kokkos_Core.hpp>

namespace Tpetra {
namespace Details {
namespace Spaces {

template <typename ExecSpace> class Slot {
public:
  using execution_space = ExecSpace;

  Slot() {
    for (int i = 0; i < static_cast<int>(Spaces::Priority::NUM_LEVELS); ++i) {
      // retrieve an RCP from the global instance manager
      instances_[i] = Spaces::space_instance<execution_space>(
          static_cast<Spaces::Priority>(i));
    }
  }

  template <Spaces::Priority priority = Spaces::Priority::medium>
  Teuchos::RCP<const execution_space> space_instance() const {
    return instances_[static_cast<int>(priority)];
  }

  Teuchos::RCP<const execution_space>
  space_instance(const Spaces::Priority &priority) const {
    switch (priority) {
    case Spaces::Priority::high:
      return space_instance<Spaces::Priority::high>();
    case Spaces::Priority::medium:
      return space_instance<Spaces::Priority::medium>();
    case Spaces::Priority::low:
      return space_instance<Spaces::Priority::low>();
    default:
      throw std::runtime_error("unexpected Tpetra Space priority");
    }
  }

private:
  Teuchos::RCP<const execution_space>
      instances_[static_cast<int>(Spaces::Priority::NUM_LEVELS)];
}; // Slot

} // namespace Spaces
} // namespace Details
} // namespace Tpetra

#endif // TPETRA_DETAILS_EXEUTIONSPACESSLOT_HPP