add a fast compilation path for static for (0..N) (#2005)

* add a fast compilation path for static for (0..N)

* Update functional2.hpp

add comment and put range applier into detail namespace

* Update functional.hpp

ditto for ck-tile

* prettify

* prettify more

* add comment

* clang-format
This commit is contained in:
Max Podkorytov
2025-04-01 12:06:25 -07:00
committed by GitHub
parent 6355ee7ca5
commit c59a8bb206
2 changed files with 48 additions and 0 deletions

View File

@@ -46,4 +46,28 @@ struct static_for
}
};
namespace detail {
template <typename T, T... Is>
struct applier
{
template <typename F>
__host__ __device__ constexpr void operator()(F f) const
{
// tweak -fbracket-depth if compilation fails. Clang default limit is 256
(f(Number<Is>{}), ...);
}
};
template <int32_t Size> // == sizeof...(Is)
using make_applier = __make_integer_seq<applier, index_t, Size>;
} // namespace detail
template <index_t N>
struct static_for<0, N, 1> : detail::make_applier<N>
{
using detail::make_applier<N>::operator();
};
} // namespace ck