profiling.hpp

Go to the documentation of this file.

// Copyright (c) Lawrence Livermore National Security, LLC and
// other Smith Project Developers. See the top-level LICENSE file for
// details.
//
// SPDX-License-Identifier: (BSD-3-Clause)
 
#pragma once
 
#include <string>
#include <sstream>
 
#include "smith/smith_config.hpp"
 
#ifdef SMITH_USE_ADIAK
#include "adiak.hpp"
#endif
 
#ifdef SMITH_USE_CALIPER
#include "caliper/cali-manager.h"
#include "caliper/cali.h"
#endif
 
#include "mpi.h"
 
#ifdef SMITH_USE_ADIAK
#define SMITH_SET_METADATA(name, data) adiak::value(name, data)
#else
#define SMITH_SET_METADATA(name, data)
#endif
 
// NOTE: The motivation behind wrapping Caliper macros to avoid conflicting macro definitions in the no-op case, and
// give downstream users the option to disable profiling Smith if it pollutes their timings.
 
#ifdef SMITH_USE_CALIPER
#define SMITH_MARK_FUNCTION CALI_CXX_MARK_FUNCTION
#define SMITH_MARK_LOOP_BEGIN(id, name) CALI_CXX_MARK_LOOP_BEGIN(id, name)
#define SMITH_MARK_LOOP_ITERATION(id, i) CALI_CXX_MARK_LOOP_ITERATION(id, i)
#define SMITH_MARK_LOOP_END(id) CALI_CXX_MARK_LOOP_END(id)
#define SMITH_MARK_BEGIN(name) CALI_MARK_BEGIN(name)
#define SMITH_MARK_END(name) CALI_MARK_END(name)
#define SMITH_MARK_SCOPE(name) CALI_CXX_MARK_SCOPE(name)
#else
// Define no-op macros in case Smith has not been configured with Caliper
#define SMITH_MARK_FUNCTION
#define SMITH_MARK_LOOP_BEGIN(id, name)
#define SMITH_MARK_LOOP_ITERATION(id, i)
#define SMITH_MARK_LOOP_END(id)
#define SMITH_MARK_BEGIN(name)
#define SMITH_MARK_END(name)
#define SMITH_MARK_SCOPE(name)
#endif
 
namespace smith::profiling {
 
void initialize([[maybe_unused]] MPI_Comm comm = MPI_COMM_WORLD, [[maybe_unused]] std::string options = "");
 
void finalize();
 
template <typename... T>
std::string concat(T... args)
{
  std::stringstream ss;
  // this fold expression is a more elegant way to implement the concatenation,
  // but nvcc incorrectly generates warning "warning: expression has no effect"
  // when using the fold expression version
  // (ss << ... << args);
  ((ss << args), ...);
  return ss.str();
}
 
}  // namespace smith::profiling