Skip to content
This repository has been archived by the owner on Mar 21, 2024. It is now read-only.

Commit

Permalink
Use inline namespace
Browse files Browse the repository at this point in the history
  • Loading branch information
gevtushenko committed Aug 12, 2022
1 parent 604abdc commit 5edc5ad
Showing 1 changed file with 57 additions and 1 deletion.
58 changes: 57 additions & 1 deletion cub/util_namespace.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -108,6 +108,60 @@
#define CUB_NS_QUALIFIER ::cub
#endif

#define CUB_COUNT_N(_1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, _12, _13, _14, _15, N, ...) N
#define CUB_COUNT(...) \
CUB_IDENTITY(CUB_COUNT_N(__VA_ARGS__, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1))
#define CUB_IDENTITY(N) N
#define CUB_APPLY(MACRO, ...) CUB_IDENTITY(MACRO(__VA_ARGS__))
#define CUB_NS_NAME1(P1) \
CUB_##P1##_NS
#define CUB_NS_NAME2(P1, P2) \
CUB_##P1##_##P2##_NS
#define CUB_NS_NAME3(P1, P2, P3) \
CUB_##P1##_##P2##_##P3##_NS
#define CUB_NS_NAME4(P1, P2, P3, P4) \
CUB_##P1##_##P2##_##P3##_##P4##_NS
#define CUB_NS_NAME5(P1, P2, P3, P4, P5) \
CUB_##P1##_##P2##_##P3##_##P4##_##P5##_NS
#define CUB_NS_NAME6(P1, P2, P3, P4, P5, P6) \
CUB_##P1##_##P2##_##P3##_##P4##_##P5##_##P6##_NS
#define CUB_NS_NAME7(P1, P2, P3, P4, P5, P6, P7) \
CUB_##P1##_##P2##_##P3##_##P4##_##P5##_##P6##_##P7##_NS
#define CUB_NS_NAME8(P1, P2, P3, P4, P5, P6, P7, P8) \
CUB_##P1##_##P2##_##P3##_##P4##_##P5##_##P6##_##P7##_##P8##_NS
#define CUB_NS_NAME9(P1, P2, P3, P4, P5, P6, P7, P8, P9) \
CUB_##P1##_##P2##_##P3##_##P4##_##P5##_##P6##_##P7##_##P8##_##P9##_NS
#define CUB_NS_NAME10(P1, P2, P3, P4, P5, P6, P7, P8, P9, P10) \
CUB_##P1##_##P2##_##P3##_##P4##_##P5##_##P6##_##P7##_##P8##_##P9##_##P10##_NS
#define CUB_NS_NAME11(P1, P2, P3, P4, P5, P6, P7, P8, P9, P10, P11) \
CUB_##P1##_##P2##_##P3##_##P4##_##P5##_##P6##_##P7##_##P8##_##P9##_##P10##_##P11##_NS
#define CUB_NS_NAME12(P1, P2, P3, P4, P5, P6, P7, P8, P9, P10, P11, P12) \
CUB_##P1##_##P2##_##P3##_##P4##_##P5##_##P6##_##P7##_##P8##_##P9##_##P10##_##P11##_##P12##_NS
#define CUB_NS_NAME13(P1, P2, P3, P4, P5, P6, P7, P8, P9, P10, P11, P12, P13) \
CUB_##P1##_##P2##_##P3##_##P4##_##P5##_##P6##_##P7##_##P8##_##P9##_##P10##_##P11##_##P12##_##P13##_NS
#define CUB_NS_NAME14(P1, P2, P3, P4, P5, P6, P7, P8, P9, P10, P11, P12, P13, P14) \
CUB_##P1##_##P2##_##P3##_##P4##_##P5##_##P6##_##P7##_##P8##_##P9##_##P10##_##P11##_##P12##_##P13##_##P14##_NS
#define CUB_NS_NAME15(P1, P2, P3, P4, P5, P6, P7, P8, P9, P10, P11, P12, P13, P14, P15) \
CUB_##P1##_##P2##_##P3##_##P4##_##P5##_##P6##_##P7##_##P8##_##P9##_##P10##_##P11##_##P12##_##P13##_##P14##_##P15##_NS
#define CUB_DISPATCH(N) CUB_NS_NAME ## N
#define CUB_NS_NAME(...) CUB_IDENTITY(CUB_APPLY(CUB_DISPATCH, CUB_COUNT(__VA_ARGS__))(__VA_ARGS__))

#if defined(CUB_DISABLE_NAMESPACE_MAGIC)
#if !defined(CUB_WRAPPED_NAMESPACE)
#error "Disabling namespace magic is unsafe without wrapping namespace"
#endif
#define CUB_INNER_NS_BEGIN
#define CUB_INNER_NS_END
#else
#if defined(_NVHPC_CUDA)
#define CUB_INNER_NS_BEGIN namespace { inline namespace CUB_NS_NAME(CUB_VERSION) {
#define CUB_INNER_NS_END }}
#else
#define CUB_INNER_NS_BEGIN inline namespace CUB_NS_NAME(CUB_VERSION, __CUDA_ARCH_LIST__) {
#define CUB_INNER_NS_END }
#endif
#endif

/**
* \def CUB_NAMESPACE_BEGIN
* This macro is used to open a `cub::` namespace block, along with any
Expand All @@ -117,7 +171,8 @@
#define CUB_NAMESPACE_BEGIN \
CUB_NS_PREFIX \
namespace cub \
{
{ \
CUB_INNER_NS_BEGIN

/**
* \def CUB_NAMESPACE_END
Expand All @@ -126,6 +181,7 @@
* This macro is defined by CUB and may not be overridden.
*/
#define CUB_NAMESPACE_END \
CUB_INNER_NS_END \
} /* end namespace cub */ \
CUB_NS_POSTFIX

Expand Down

0 comments on commit 5edc5ad

Please sign in to comment.