c++ - 具有 map 和原子操作的 CUDA

Question

据我所知，我可以在 CUDA 设备代码中使用 C++ 模板。那么，如果我使用 map 创建字典，插入新值的操作会是原子操作吗？

我想计算某个值出现的次数，即创建一个包含代码概率的代码字典。

谢谢

苹果电脑

Answer 1

您不能在设备代码中使用 STL。你可以查看 thrust对于类似的功能(特别检查实验命名空间)。

模板在设备代码中很好，CUDA C 目前支持相当多的 C++ 特性，尽管一些重要的特性，如虚函数和异常还不可能(并且只能在 Fermi 硬件上实现)。

如果您决定自己实现它，您可以使用 atomicAdd() 内在函数来获得原子操作，查看 CUDA Programming Guide获取更多信息。