common.hpp

#ifndef __COMMON_EXPORT_HPP__
#define __COMMON_EXPORT_HPP__
 
#include <Kokkos_Core_fwd.hpp>
#include <string_view>
 
using ComputeSpace = Kokkos::DefaultExecutionSpace;
using HostSpace = Kokkos::DefaultHostExecutionSpace;
 
#ifdef ENABLE_KOKKOS_PROFILING
#  include <Kokkos_Profiling_ScopedRegion.hpp>
#  define PROFILE_SECTION(__label_section__)                                   \
    Kokkos::Profiling::ScopedRegion region(__label_section__);
#else
#  define PROFILE_SECTION(__label_section__) ;
#endif
 
#define EIGEN_INDEX(__VALUE__) static_cast<int>(__VALUE__)
 
#ifndef NDEBUG
#  include <iostream>
#  include <source_location>
class Canary
{
public:
  Canary() = delete;
  Canary(const Canary&) = delete;
  Canary(Canary&&) = delete;
  Canary& operator=(const Canary&) = delete;
  Canary& operator=(Canary&&) = delete;
 
  explicit Canary(std::string_view lbl, std::source_location location)
      : _lbl(lbl)
  {
    // std::cout << "\033[1;31m" << location.function_name() << "\033[0m " <<
    // lbl << std::endl;
    std::cout << location.function_name() << ": " << lbl << std::endl;
  }
 
  ~Canary()
  {
    std::cout << "END " << _lbl << std::endl;
  }
 
private:
  std::string _lbl;
};
#  define MkCanary(x) Canary(x, std::source_location());
#else
#  define MkCanary(x)
#endif
 
#endif //__COMMON_EXPORT_HPP__