Tpetra_Details_KokkosCounter.hpp

Go to the documentation of this file.

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

// clang-format off
#ifndef TPETRA_DETAILS_KOKKOS_COUNTER_HPP
#define TPETRA_DETAILS_KOKKOS_COUNTER_HPP


#include <string>
#include <Teuchos_FancyOStream.hpp>

namespace Tpetra {
namespace Details {

namespace DeepCopyCounter {
  void start();

  void reset();

  void stop();

  size_t get_count_same_space();

  size_t get_count_different_space();

}

namespace FenceCounter {
  void start();

  void reset();

  void stop();

  size_t get_count_instance(const std::string & device);

  size_t get_count_global(const std::string & device);
}

// clang-format on

namespace KokkosRegionCounter {
void start();

void reset();

void stop();

size_t get_count_region_contains(const std::string &substr);

void dump_regions(std::ostream &os);
void dump_regions(Teuchos::FancyOStream &os);
} // namespace KokkosRegionCounter

// clang-format off



} // namespace Details
} // namespace Tpetra

#endif // TPETRA_DETAILS_KOKKOS_COUNTER_HPP