The ROCm software primatives library is rocPRIM [1] and the ROCm equivalent to Thrust is rocThrust [2]. Though, if you're a user of CUB, it might be easier to use hipCUB [3] to support both platforms.
That's HIP, and you're putting your faith in ongoing support, but, it's cool that those libraries got ported. I bet my old thesis would probably HIPify pretty well.
[1]: https://github.com/ROCmSoftwarePlatform/rocPRIM
[2]: https://github.com/ROCmSoftwarePlatform/rocThrust
[3]: https://github.com/ROCmSoftwarePlatform/hipCUB