C++ Primitive Libraries

Applies to Linux and Windows

2023-09-15

4 min read time

ROCm template libraries for algorithms are as follows:

rocPRIM

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.

• Documentation

• GitHub

• Changelog

• Examples

rocThrust

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.

• Documentation

• GitHub

• Changelog

• Examples

hipCUB

hipCUB is a template library of algorithm primitives with a CUB-compatible interface. It’s back-end is rocPRIM.

• Documentation

• GitHub

• Changelog

• Examples

hipTensor

hipTensor is AMD’s C++ library for accelerating tensor primitives based on the composable kernel library, through general purpose kernel languages, like HIP C++.

• Documentation

• GitHub