英特尔 MKL 稀疏 QR 求解 C++ 返回未初始化错误



尝试使用mkl_sparse_s_qr_solve时,我收到所有0的结果和错误状态SPARSE_STATUS_NOT_INITIALIZED这意味着句柄/矩阵为空。

我尝试通读文档并打印用于实例化求解所需的 CSR 稀疏矩阵的所有数组,并且所有数组都包含正确的值。

int main() {
    std::vector<int> i_b(22);
    std::vector<int> i_e(22);
    // j, v, and b vectors are just examples.
    // Regardless of the resulting numbers of the solve, I just
    // cannot get the SPARSE_STATUS_NOT_INITIALIZED error to stop
    // for the qr solver.
    std::vector<int> j(44, 0);
    std::vector<float> v(44, 1.0);
    std::vector<float> b(22, 1.0);
    {
        struct IncGenerator {
            int current_;
            IncGenerator(int start) : current_(start) {}
            int operator() () {
                current_ += 2;
                return current_;
            }
        };
        // Fill i_b with {0, 2, 4, ..., 42}
        IncGenerator g(-2);
        std::generate(i_b.begin(), i_b.end(), g);
        // Fill i_e with {2, 4, 6, ..., 44}
        IncGenerator f(0);
        std::generate(i_e.begin(), i_e.end(), f);
    }
    // ...
    // j, v, and b arrays are all the correct values
    // confirmed. The sparse A matrix should have 2 values
    // per row, with 22 rows, and 15 columns.
    int out;
    sparse_matrix_t A;
    out = mkl_sparse_s_create_csr(&A, SPARSE_INDEX_BASE_ZERO, 22, 15, &i_b[0], &i_e[0], &j[0], &v[0]);
    switch (out) {
    case SPARSE_STATUS_SUCCESS:
        std::cout << "Successfully created matrix!" << std::endl;
        break;
    case SPARSE_STATUS_NOT_INITIALIZED:
        std::cout << "Not initialized." << std::endl;
        break;
    case SPARSE_STATUS_ALLOC_FAILED:
        std::cout << "Internal memory allocation failed." << std::endl;
        break;
    default:
        std::cout << "Unknown." << std::endl;
        break;
    }
    std::vector<float> X(22 * 15);
    out = mkl_sparse_s_qr_solve(SPARSE_OPERATION_NON_TRANSPOSE, A, NULL, SPARSE_LAYOUT_COLUMN_MAJOR, 1, &X[0], 15, &asv[0], 22);
    switch (out) {
    case SPARSE_STATUS_SUCCESS:
        std::cout << "Successfully solved!" << std::endl;
        break;
    case SPARSE_STATUS_NOT_INITIALIZED:
        std::cout << "Not initialized." << std::endl;
        break;
    case SPARSE_STATUS_ALLOC_FAILED:
        std::cout << "Internal memory allocation failed." << std::endl;
        break;
    default:
        std::cout << "Unknown." << std::endl;
        break;
    }
    return 0;
}

因此,由于某种原因,我无法用A解决,因为它要么认为 A 为空,要么其他东西未初始化。我不认为A是空的(我想检查但没有方便的方法来打印A(,因为矩阵的初始化A返回为成功的操作(我唯一有点怀疑的是行开头i_b和行结束i_e索引(。

任何人都可以提供一些指导吗?

这不是

你应该使用mkl_sparse_?_qr_solve的方式。稀疏系统分 3 个步骤(阶段(解决:

  1. 重新安排。
  2. 因式分解。
  3. 解决。

首先,您必须调用 mkl_sparse_qr_reorder ,然后是 mkl_sparse_?_qr_factorize ,然后才调用 mkl_sparse_?_qr_solve

尝试在mkl_sparse_?_qr_solve之前插入以下代码:

struct matrix_descr descr;
descr.type = SPARSE_MATRIX_TYPE_GENERAL;
out = mkl_sparse_qr_reorder(A, descr);
switch (out) { ... }
out = mkl_sparse_?_qr_factorize(A, NULL);
switch (out) { ... }

或者只需使用mkl_sparse_?_qr即可在一次通话中为您完成所有 3 个步骤。将流程分为三个步骤,为您提供了更大的自由度。例如,如果要用相同的A求解多个系统,则可以通过调用mkl_sparse_qr_reorder并仅mkl_sparse_?_qr_factorize一次来节省时间。

没有直接关系,但不要使用int而不是MKL_INT。定义MKL_ILP64时,MKL_INT不是int,而是long long int

最新更新