C++ Primitive Libraries

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++.