C++ Primitive Libraries#
Applies to Linux and Windows
2023-09-15
4 min read time
ROCm template libraries for algorithms are as follows:
rocPRIM is an AMD GPU optimized template library of algorithm primitives, like transforms, reductions, scans, etc. It also serves as a common back-end for similar libraries found inside ROCm.
rocThrust is a template library of algorithm primitives with a Thrust-compatible interface. Their CPU back-ends are identical, while the GPU back-end calls into rocPRIM.
hipCUB is a template library of algorithm primitives with a CUB-compatible interface. It’s back-end is rocPRIM.
hipTensor is AMD’s C++ library for accelerating tensor primitives based on the composable kernel library, through general purpose kernel languages, like HIP C++.