NVPL Sparse 通用 API#

  • 通用 API 提供了 NVPL Sparse 的标准接口层。它们允许以灵活的方式计算最常见的稀疏线性代数运算，例如稀疏矩阵-向量 (SpMV) 和稀疏矩阵-矩阵乘法 (SpMM)。这些 API 具有以下功能和特性

    • 设置矩阵数据布局、批次数量和存储格式（例如，CSR、COO 等）。

    • 设置输入/输出/计算数据类型。这也允许混合数据类型计算。

    • 设置稀疏向量/矩阵索引的类型（例如，32 位、64 位）。

    • 选择计算的算法。

    • 保证内部操作的外部内存。

    • 提供对输入矩阵和向量的广泛一致性检查。这包括验证大小、数据类型、布局、允许的操作等。

    • 为向量和矩阵输入提供常量描述符，以支持 const 安全接口并保证 API 不会修改其输入。