cuda - CUDA 是否保证 float 为 4 个字节？

Question

我想使用 cudaMemcpy 创建一个向量 (1,...,1) 这样我就可以做一些事情，比如对矩阵的行/列求和或将向量添加到矩阵与 CUBLAS。代码会在不同的平台上运行，所以

• 我如何保证我始终使用 4 字节 float ？
• GPU 上是否有针对数据类型的 sizeof 函数，或者设备是否始终使用与主机相同的数据类型规范？

Answer 1

首先回答你的第二个问题，设备总是使用与主机编译器相同的数据类型规范。因此，您可以在主机上使用 sizeof(...) 来确定相关大小。如果不是这样，很多东西都会坏掉。

要回答您的第一个问题，那么，我们只需要问，在 CUDA 支持的主机端编译器中，float 表示总是 32 位吗？答案是yes .

顺便说一句，请注意，这通常适用于大多数平台。查找具有非 32 位 float 的系统 is difficult .但据我所知，没有一般的 C 或 C++ 要求 float 是 32 位。其他人可能会证明我错了。