metaprogramming.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 <tuple>
#include <utility>
#include <type_traits>
 
#include "smith/infrastructure/accelerator.hpp"
 
template <int I, int... n>
constexpr auto get(std::integer_sequence<int, n...>)
{
  constexpr int values[sizeof...(n)] = {n...};
  return values[I];
}
 
namespace detail {
 
template <typename T>
struct always_false : std::false_type {};
 
template <int i>
struct integral_constant {
  SMITH_HOST_DEVICE constexpr operator int() { return i; }
};
 
SMITH_SUPPRESS_NVCC_HOSTDEVICE_WARNING
template <typename lambda, int... i>
SMITH_HOST_DEVICE constexpr void for_constexpr(const lambda& f, integral_constant<i>... args)
{
  f(args...);
}
 
SMITH_SUPPRESS_NVCC_HOSTDEVICE_WARNING
template <int... n, typename lambda, typename... arg_types>
SMITH_HOST_DEVICE constexpr void for_constexpr(const lambda& f, std::integer_sequence<int, n...>, arg_types... args)
{
  (detail::for_constexpr(f, args..., integral_constant<n>{}), ...);
}
 
}  // namespace detail
 
template <int... n, typename lambda>
SMITH_HOST_DEVICE constexpr void for_constexpr(const lambda& f)
{
  detail::for_constexpr(f, std::make_integer_sequence<int, n>{}...);
}