库组织和功能

NVPL Sparse 库按以下 API 集组织

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

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

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

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

  • 选择计算的算法。

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

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

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