normalizeOperand.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 RAJA_pattern_tensor_ET_normalizeOperand_HPP
#define RAJA_pattern_tensor_ET_normalizeOperand_HPP
 
#include "RAJA/config.hpp"
 
#include "RAJA/util/macros.hpp"
 
#include "RAJA/pattern/tensor/internal/TensorRef.hpp"
 
namespace RAJA
{
namespace internal
{
namespace expt
{
 
 
class TensorRegisterConcreteBase;
 
namespace ET
{
class TensorExpressionConcreteBase;
 
template<typename RHS, typename enable = void>
struct NormalizeOperandHelper;
 
/*
 * For TensorExpression nodes, we just return them as-is.
 */
template<typename RHS>
struct NormalizeOperandHelper<
    RHS,
    typename std::enable_if<
        std::is_base_of<TensorExpressionConcreteBase, RHS>::value>::type>
{
  using return_type = RHS;
 
  RAJA_INLINE
 
  RAJA_HOST_DEVICE
  static constexpr return_type normalize(RHS const& rhs) { return rhs; }
};
 
template<typename RHS>
RAJA_INLINE RAJA_HOST_DEVICE auto normalizeOperand(RHS const& rhs) ->
    typename NormalizeOperandHelper<RHS>::return_type
{
  return NormalizeOperandHelper<RHS>::normalize(rhs);
}
 
template<typename RHS>
using normalize_operand_t = typename NormalizeOperandHelper<RHS>::return_type;
 
 
}  // namespace ET
 
}  // namespace expt
}  // namespace internal
 
}  // namespace RAJA
 
 
#endif