Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

make the upper limit on TMP loop unrolling configurable #2971

Merged
merged 4 commits into from
Dec 1, 2024
Merged

make the upper limit on TMP loop unrolling configurable #2971

Changes from 2 commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
55a9180
make the upper limit on TMP loop unrolling configurable
ericniebler Nov 27, 2024
793a274
fix dunderheaded mistake
ericniebler Nov 27, 2024
ce60894
Merge remote-tracking branch 'origin/main' into configurable-type-alg…
ericniebler Nov 30, 2024
a350150
drop the TMP PP unrolling limit on nvcc and nvhpc whose parsers are slow
ericniebler Nov 30, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
32 changes: 18 additions & 14 deletions libcudacxx/include/cuda/std/__type_traits/type_list.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,10 @@
//! For the purpose of this file, a "trait type" is a class type with a nested
//! type alias named \c type.

#if !defined(_CCCL_META_UNROLL_LIMIT)
# define _CCCL_META_UNROLL_LIMIT 16
#endif

_LIBCUDACXX_BEGIN_NAMESPACE_STD

#ifndef _CCCL_DOXYGEN_INVOKED // Do not document
Expand Down Expand Up @@ -502,7 +506,7 @@ struct __type_index_small_size_fn;
using __call _CCCL_NODEBUG_ALIAS = _Ty; \
};

_CCCL_PP_REPEAT_REVERSE(16, _M1)
_CCCL_PP_REPEAT_REVERSE(_CCCL_META_UNROLL_LIMIT, _M1)

# undef _M0
# undef _M1
Expand All @@ -524,7 +528,7 @@ struct __type_index_select_fn<true> // Fast implementation for smaller indices
} // namespace __detail

template <class _Ip, class... _Ts>
using __type_index = __type_call<__detail::__type_index_select_fn<(_Ip::value < 16)>, _Ip, _Ts...>;
using __type_index = __type_call<__detail::__type_index_select_fn<(_Ip::value < _CCCL_META_UNROLL_LIMIT)>, _Ip, _Ts...>;

template <size_t _Ip, class... _Ts>
using __type_index_c = __type_index<integral_constant<size_t, _Ip>, _Ts...>;
Expand Down Expand Up @@ -764,17 +768,17 @@ struct _CCCL_TYPE_VISIBILITY_DEFAULT __type_fold_left_fn;
using __call _CCCL_NODEBUG_ALIAS = _CCCL_PP_REPEAT(_N, _M1) _State _CCCL_PP_REPEAT(_N, _M3); \
};

_CCCL_PP_REPEAT_REVERSE(17, _LIBCUDACXX_TYPE_LIST_FOLD_RIGHT)
_CCCL_PP_REPEAT_REVERSE(_CCCL_PP_INC(_CCCL_META_UNROLL_LIMIT), _LIBCUDACXX_TYPE_LIST_FOLD_RIGHT)

template <size_t _Np>
struct _CCCL_TYPE_VISIBILITY_DEFAULT __type_fold_right_fn
{
template <class _Fn, class _State _CCCL_PP_REPEAT(16, _M0), class... _Rest>
using __call _CCCL_NODEBUG_ALIAS =
__type_call_indirect<__type_fold_right_fn<_Np - 16>,
_Fn,
__type_call<__type_fold_right_fn<16>, _Fn, _State _CCCL_PP_REPEAT(16, _M2)>,
_Rest...>;
template <class _Fn, class _State _CCCL_PP_REPEAT(_CCCL_META_UNROLL_LIMIT, _M0), class... _Rest>
using __call _CCCL_NODEBUG_ALIAS = __type_call_indirect<
__type_fold_right_fn<_Np - _CCCL_META_UNROLL_LIMIT>,
_Fn,
__type_call<__type_fold_right_fn<_CCCL_META_UNROLL_LIMIT>, _Fn, _State _CCCL_PP_REPEAT(_CCCL_META_UNROLL_LIMIT, _M2)>,
_Rest...>;
};

template <class _Init, class _Fn>
Expand All @@ -792,17 +796,17 @@ struct _CCCL_TYPE_VISIBILITY_DEFAULT __type_fold_right_select_fn
using __call _CCCL_NODEBUG_ALIAS = _CCCL_PP_REPEAT(_N, _M1) _State _CCCL_PP_REPEAT(_N, _M4, _N, _CCCL_PP_DEC); \
};

_CCCL_PP_REPEAT_REVERSE(17, _LIBCUDACXX_TYPE_FOLD_LEFT)
_CCCL_PP_REPEAT_REVERSE(_CCCL_PP_INC(_CCCL_META_UNROLL_LIMIT), _LIBCUDACXX_TYPE_FOLD_LEFT)

template <size_t _Np>
struct _CCCL_TYPE_VISIBILITY_DEFAULT __type_fold_left_fn
{
template <class _Fn, class _State _CCCL_PP_REPEAT(16, _M0), class... _Rest>
template <class _Fn, class _State _CCCL_PP_REPEAT(_CCCL_META_UNROLL_LIMIT, _M0), class... _Rest>
using __call _CCCL_NODEBUG_ALIAS =
__type_call<__type_fold_left_fn<16>,
__type_call<__type_fold_left_fn<_CCCL_META_UNROLL_LIMIT>,
_Fn,
__type_call_indirect<__type_fold_left_fn<_Np - 16>, _Fn, _State, _Rest...> //
_CCCL_PP_REPEAT(16, _M2, 0, _CCCL_PP_INC)>;
__type_call_indirect<__type_fold_left_fn<_Np - _CCCL_META_UNROLL_LIMIT>, _Fn, _State, _Rest...> //
_CCCL_PP_REPEAT(_CCCL_META_UNROLL_LIMIT, _M2, 0, _CCCL_PP_INC)>;
};

template <class _Init, class _Fn>
Expand Down
Loading