Skip to content

Commit c53404e

Browse files
Add missed specializations of the new aligned vector types to cub (#5264) (#5271)
Without those, our tests will fail because of the fallback implementation of `numeric_limits` setting everything to 0. Fixes nvbug5391162 Co-authored-by: Bernhard Manfred Gruber <[email protected]>
1 parent aa9ea15 commit c53404e

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

c2h/generators.cu

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -619,6 +619,7 @@ VEC_SPECIALIZATION(long, 4);
619619
_CCCL_SUPPRESS_DEPRECATED_POP
620620
# if _CCCL_CTK_AT_LEAST(13, 0)
621621
VEC_SPECIALIZATION(long, 4, _16a);
622+
VEC_SPECIALIZATION(long, 4, _32a);
622623
# endif // _CCCL_CTK_AT_LEAST(13, 0)
623624

624625
// VEC_SPECIALIZATION(ulong, 2);
@@ -632,6 +633,7 @@ VEC_SPECIALIZATION(longlong, 4);
632633
_CCCL_SUPPRESS_DEPRECATED_POP
633634
# if _CCCL_CTK_AT_LEAST(13, 0)
634635
VEC_SPECIALIZATION(longlong, 4, _16a);
636+
VEC_SPECIALIZATION(longlong, 4, _32a);
635637
# endif // _CCCL_CTK_AT_LEAST(13, 0)
636638

637639
VEC_SPECIALIZATION(ulonglong, 2);
@@ -641,6 +643,7 @@ VEC_SPECIALIZATION(ulonglong, 4);
641643
_CCCL_SUPPRESS_DEPRECATED_POP
642644
# if _CCCL_CTK_AT_LEAST(13, 0)
643645
VEC_SPECIALIZATION(ulonglong, 4, _16a);
646+
VEC_SPECIALIZATION(ulonglong, 4, _32a);
644647
# endif // _CCCL_CTK_AT_LEAST(13, 0)
645648

646649
VEC_SPECIALIZATION(float, 2);
@@ -654,6 +657,7 @@ VEC_SPECIALIZATION(double, 4);
654657
_CCCL_SUPPRESS_DEPRECATED_POP
655658
# if _CCCL_CTK_AT_LEAST(13, 0)
656659
VEC_SPECIALIZATION(double, 4, _16a);
660+
VEC_SPECIALIZATION(double, 4, _32a);
657661
# endif // _CCCL_CTK_AT_LEAST(13, 0)
658662

659663
template <typename VecType, typename Type>
@@ -685,6 +689,7 @@ VEC_GEN_MOD_SPECIALIZATION(uchar3, unsigned char);
685689
_CCCL_SUPPRESS_DEPRECATED_PUSH
686690
# if _CCCL_CTK_AT_LEAST(13, 0)
687691
VEC_GEN_MOD_SPECIALIZATION(ulonglong4_16a, unsigned long long);
692+
VEC_GEN_MOD_SPECIALIZATION(ulonglong4_32a, unsigned long long);
688693
# else
689694
VEC_GEN_MOD_SPECIALIZATION(ulonglong4, unsigned long long);
690695
# endif // _CCCL_CTK_AT_LEAST(13, 0)

c2h/include/c2h/test_util_vec.h

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -366,6 +366,17 @@ _CCCL_SUPPRESS_DEPRECATED_PUSH
366366
C2H_VEC_TRAITS_OVERLOAD(double, double)
367367
_CCCL_SUPPRESS_DEPRECATED_POP
368368

369+
# if _CCCL_CTK_AT_LEAST(13, 0)
370+
C2H_VEC_TRAITS_OVERLOAD_IMPL(long4_16a, long, 4)
371+
C2H_VEC_TRAITS_OVERLOAD_IMPL(long4_32a, long, 4)
372+
C2H_VEC_TRAITS_OVERLOAD_IMPL(ulong4_16a, unsigned long, 4)
373+
C2H_VEC_TRAITS_OVERLOAD_IMPL(ulong4_32a, unsigned long, 4)
374+
C2H_VEC_TRAITS_OVERLOAD_IMPL(longlong4_16a, long long, 4)
375+
C2H_VEC_TRAITS_OVERLOAD_IMPL(longlong4_32a, long long, 4)
376+
C2H_VEC_TRAITS_OVERLOAD_IMPL(ulonglong4_16a, unsigned long long, 4)
377+
C2H_VEC_TRAITS_OVERLOAD_IMPL(ulonglong4_32a, unsigned long long, 4)
378+
# endif // _CCCL_CTK_AT_LEAST(13, 0)
379+
369380
# undef C2H_VEC_TRAITS_OVERLOAD
370381
# undef C2H_VEC_TRAITS_OVERLOAD_IMPL
371382
# undef REPEAT_TO_LIST_1

0 commit comments

Comments
 (0)