cudaMemset()-它设置字节还是整数

Question

来自在线文档：

cudaError_t cudaMemset (void * devPtr, int value, size_t count )

用常量字节值填充devPtr指向的内存区域的第一个计数字节。

参数：devPtr-指向设备内存的指针value—为指定内存的每个字节设置的值count-设置的大小(以字节为单位)

此描述似乎不正确，因为：

int *dJunk;
cudaMalloc((void**)&dJunk, 32*(sizeof(int));
cudaMemset(dJunk, 0x12, 32);

将所有32个整数设置为0x12，而不是0x12121212。(Int vs Byte)

描述中谈到了设置字节。计数和值以字节为单位进行描述。注意count的类型为size_t，value的类型为int。例如，将字节大小设置为int值。

程序指南中没有提到cudaMemset()。我不得不假设我看到的行为是正确的，而文档是糟糕的。

有更好的文档来源吗？(哪里？)
是否支持其他类型？即float *dJunk;是否有效？其他人？

Answer 1

文档是正确的，您对cudaMemset功能的解释是错误的。该函数确实设置了字节值。您的示例将前32个字节设置为0x12，而不是将所有32个整数都设置为0x12，即：

#include <cstdio>
int main(void)
{
const int n = 32;
const size_t sz = size_t(n) * sizeof(int);
int *dJunk;
cudaMalloc((void**)&dJunk, sz);
cudaMemset(dJunk, 0, sz);
cudaMemset(dJunk, 0x12, 32);
int *Junk = new int[n];
cudaMemcpy(Junk, dJunk, sz, cudaMemcpyDeviceToHost);
for(int i=0; i<n; i++) {
fprintf(stdout, "%d %xn", i, Junk[i]);
}
cudaDeviceReset();
return 0;
}

产生

$ nvcc memset.cu 
$ ./a.out 
0 12121212
1 12121212
2 12121212
3 12121212
4 12121212
5 12121212
6 12121212
7 12121212
8 0
9 0
10 0
11 0
12 0
13 0
14 0
15 0
16 0
17 0
18 0
19 0
20 0
21 0
22 0
23 0
24 0
25 0
26 0
27 0
28 0
29 0
30 0
31 0

即。所有128个字节都设置为0，然后第一个32个字节设置为0x12。正如文件所述。