ipiv上的双重释放或损坏（输出）magma_getrs_gpu

Question

我有以下代码：

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出了什么问题。

有什么主意吗？

Answer 1

我终于找到了解决方案。与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);