kernel_name.hpp

Go to the documentation of this file.

//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~//
// Copyright (c) Lawrence Livermore National Security, LLC and other
// RAJA Project Developers. See top-level LICENSE and COPYRIGHT
// files for dates and other details. No copyright assignment is required
// to contribute to RAJA.
//
// SPDX-License-Identifier: (BSD-3-Clause)
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~//
 
#ifndef HIP_KERNELNAME_HPP
#define HIP_KERNELNAME_HPP
 
#if defined(RAJA_HIP_ACTIVE)
 
#include "RAJA/policy/hip/MemUtils_HIP.hpp"
#include "RAJA/pattern/params/kernel_name.hpp"
 
#if defined(RAJA_ENABLE_ROCTX)
#include "hip/hip_runtime_api.h"
#include "roctx.h"
#endif
 
#if defined(RAJA_ENABLE_CALIPER)
#include "RAJA/util/CaliperPlugin.hpp"
#endif
 
namespace RAJA
{
namespace expt
{
namespace detail
{
 
// Init
template<typename EXEC_POL>
camp::concepts::enable_if<RAJA::type_traits::is_hip_policy<EXEC_POL>>
param_init(EXEC_POL const&,
           RAJA::detail::Name& kn,
           const RAJA::hip::detail::hipInfo&)
{
#if defined(RAJA_ENABLE_ROCTX)
  if (kn.name != nullptr
#if defined(RAJA_ENABLE_CALIPER)
      && RAJA::util::RAJA_caliper_profile == false
#endif
  )
  {
    roctxRangePush(kn.name);
  }
#else
  RAJA_UNUSED_VAR(kn);
#endif
}
 
// Combine
template<typename EXEC_POL>
RAJA_HOST_DEVICE camp::concepts::enable_if<
    RAJA::type_traits::is_hip_policy<EXEC_POL>>
param_combine(EXEC_POL const&, RAJA::detail::Name&)
{}
 
// Resolve
template<typename EXEC_POL>
camp::concepts::enable_if<RAJA::type_traits::is_hip_policy<EXEC_POL>>
param_resolve(EXEC_POL const&,
              RAJA::detail::Name& kn,
              const RAJA::hip::detail::hipInfo&)
{
#if defined(RAJA_ENABLE_ROCTX)
  if (kn.name != nullptr
#if defined(RAJA_ENABLE_CALIPER)
      && RAJA::util::RAJA_caliper_profile == false
#endif
  )
  {
    roctxRangePop();
  }
#else
  RAJA_UNUSED_VAR(kn);
#endif
}
 
}  //  namespace detail
}  //  namespace expt
}  //  namespace RAJA
 
#endif
 
#endif  //  NEW_REDUCE_HIP_REDUCE_HPP