ipiv上的双重释放或损坏(输出)magma_getrs_gpu

时间:2019-05-02 19:58:44

标签: c++ lapack magma

我有以下代码:

magma_int_t *h_ipiv;
magma_imalloc_cpu( &h_ipiv,  k);

for (i=0;i<k;i++){
    h_ipiv[i] = i;
}
magma_dgemm(MagmaTrans, MagmaNoTrans, N, n, m, 1.0, d_G2, m, d_A2, m, 0.0, d_QA, N, queue);

magma_dgemm(MagmaTrans, MagmaNoTrans, N, n, m, 1.0, d_G1, m, d_A1, m, -1.0, d_QA, N, queue);

magma_int_t info_getrs;
magma_dgetrs_gpu(MagmaTrans, Ngaps, n, d_M, N, h_ipiv, d_QA, N, &info_getrs);

在最后一行,我得到一个错误double free or corruption (out): 0x000000001dd18540。我检查了magma_dgemm中的数组,它们似乎是正确的。大小也是正确的,因为我在cpu lapack版本上有此代码,并且它们是相同的。因此错误似乎出在h_ipiv上,但是我不知道h_ipiv出了什么问题。

有什么主意吗?

1 个答案:

答案 0 :(得分:0)

我终于找到了解决方案。与lapack不同,在岩浆中,要为ipiv创建dgetrs_gpu,我们必须在dgetrf_gpu之前使用。

因此,解决方案是在dgetrs_gpu函数之前添加以下几行:

magma_int_t info_getrf;
magma_dgetrf_gpu(Ngaps, Ngaps, d_M, Ngaps, h_ipiv, &info_getrf);