c++ - 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);