我有一个cblas和lapack代码,我试图“转换”成magma代码,以利用GPU。这是我的CPU代码,运行良好:
cblas_dsyrk(CblasColMajor,CblasUpper,CblasTrans,n,m,1.0,A2,m, 0.0,C_theta,n);
cblas_dsyrk(CblasColMajor,CblasUpper,CblasTrans,n,m,1.0,A1,m,-1.0,C_theta,n);
clapack_dpotrf(CblasColMajor,CblasUpper,n,C_theta,n);现在,对于magma GPU版本,我这样做了:
magma_dmalloc(&d_A1, m*n);
magma_dmalloc(&d_A2, m*n);
magma_dmalloc(&d_C_theta, n*n);
magma_dsetmatrix(m ,n, A1, m, d_A1, m, queue);
magma_dsetmatrix(m ,n, A2, m, d_A2, m, queue);
magma_dsetmatrix(n ,n, C_theta, n, d_C_theta, n, queue);
magma_dsyrk(MagmaUpper, MagmaTrans, n, m, 1.0, d_A2, m, 0.0, d_C_theta, n, queue);
magma_dsyrk(MagmaUpper, MagmaTrans, n, m, 1.0, d_A1, m, -1.0, d_C_theta, n, queue);
magma_int_t *info_potrf;
magma_dpotrf_gpu(MagmaUpper, n, d_C_theta, n, info_potrf);由于某些原因,最后一行出现了分割错误。我在这里做错了什么?这一切似乎都是正确的。
发布于 2019-04-10 22:32:07
@RobertCrovella回答了这个问题。
需要删除星号以便为info_potrf变量分配内存,在第二行,需要添加&以使用变量address。这是正确的方式:
magma_int_t info_potrf;
magma_dpotrf_gpu(MagmaUpper, n, d_C_theta, n, &info_potrf);https://stackoverflow.com/questions/55604276
复制相似问题