hipCUB is a thin header-only wrapper library on top of rocPRIM or CUB. It enables developers to port project using CUB library to the HIP layer and to run them on AMD hardware. In the ROCm environment, hipCUB uses rocPRIM library as the backend, however, on CUDA platforms it uses CUB instead.
When using hipCUB you should only include
When rocPRIM is used as backend
When CUB is used as backend
Backends are automaticaly selected based on platform detected by HIP layer (
hipCUB with the rocPRIM backend may not support all function and features CUB has because of the differences between the ROCm (HIP) platform and the CUDA platform.
Not-supported features and differences:
Functions, classes and macros which are not in the public API or not documented are not supported.
Device-wide primitives can’t be called from kernels (dynamic parallelism is not supported in HIP on ROCm).
DeviceSpmvis not supported.
TexRefInputIteratorare not supported.
CacheStoreModifiercache modifiers are not supported.
ThreadStorefunctions are not supported.
Storage management and debug functions:
_CubLogmacros are not supported.
ThreadTrap- not supported.
Warp thread masks (when used) are 64-bit unsigned integers.
member_maskinput argument is ignored in
member_maskare ignored in
CastOp- not supported.