10

来自在线文档：

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

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

参数： devPtr - 指向设备内存值的指针 - 为指定内存计数的每个字节设置的值 - 要设置的大小（以字节为单位）

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

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

将所有 32 个整数设置为 0x12，而不是 0x12121212。（整数与字节）

描述谈到设置字节。Count 和 Value 以字节为单位进行描述。注意 count 是size_t类型， value 是 int 类型。即，将字节大小设置为 int 值。

cudaMemset() 在 prog 指南中没有提到。我必须假设我看到的行为是正确的，并且文档很糟糕。

那里有更好的文档来源吗？（在哪里？）
是否支持其他类型？即会float *dJunk;工作吗？其他的？

4

1 回答 1

19

文档是正确的，而您对所做操作的解释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 %x\n", 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

IE。所有 128 个字节设置为 0，然后前 32 个字节设置为0x12. 完全按照文档中的描述。

于 2012-11-14T21:43:44.120 回答