我遇到了一个问题,不明白为什么使用CUBLAS查找一组double型数据的最大值和最小值的函数无法正常工作。
代码如下:
void findMaxAndMinGPU(double* values, int* max_idx, int* min_idx, int n)
{
double* d_values;
cublasHandle_t handle;
cublasStatus_t stat;
safecall( cudaMalloc((void**) &d_values, sizeof(double) * n), "cudaMalloc (d_values) in findMaxAndMinGPU");
safecall( cudaMemcpy(d_values, values, sizeof(double) * n, cudaMemcpyHostToDevice), "cudaMemcpy (h_values > d_values) in findMaxAndMinGPU");
cublasCreate(&handle);
stat = cublasIdamax(handle, n, d_values, sizeof(double), max_idx);
if (stat != CUBLAS_STATUS_SUCCESS)
printf("Max failed\n");
stat = cublasIdamin(handle, n, d_values, sizeof(double), min_idx);
if (stat != CUBLAS_STATUS_SUCCESS)
printf("min failed\n");
cudaFree(d_values);
cublasDestroy(handle);
}
在这里,values是要搜索的数值。max_idx和min_idx是找到的数值在values中的索引。 CUBLAS调用的结果似乎很随机,并输出错误的索引。
有没有人能够给我一个好的答案来解决我的问题?我现在有点难过 :(
max_idx
和min_idx
初始化为int
类型,而不是int *
类型,并通过引用&max_idx
传递给cublas。 - Vanwaril