稠密向量 API#

本节介绍用于稠密向量描述符的 NVPL Sparse 辅助函数。

有关存储格式的详细描述,请参阅稠密向量格式部分。


nvpl_sparse_create_dn_vec()#

nvpl_sparse_status_t
nvpl_sparse_create_dn_vec(nvpl_sparse_dn_vec_descr_t* dn_vec_descr,
                        int64_t                       size,
                        void*                         values,
                        nvpl_sparse_data_type_t       value_type)

nvpl_sparse_status_t
nvpl_sparse_create_const_dn_vec(nvpl_sparse_const_dn_vec_descr_t* dn_vec_descr,
                              int64_t                             size,
                              const void*                         values,
                              nvpl_sparse_data_type_t             value_type)

此函数初始化稠密向量描述符 dn_vec_descr

参数。

输入/输出

含义

dn_vec_descr

输出

稠密向量描述符

size

输入

稠密向量的大小

values

输入

稠密向量的值。包含 size 个元素的数组

value_type

输入

枚举器,指定 values 的数据类型

nvpl_sparse_create_dn_vec() 具有以下约束

  • values 必须与 value_type 指定的数据类型的大小对齐。有关数据类型的描述,请参阅 nvpl_sparse_data_type_t

有关返回状态的描述,请参阅 nvpl_sparse_status_t


nvpl_sparse_destroy_dn_vec()#

nvpl_sparse_status_t
nvpl_sparse_destroy_dn_vec(nvpl_sparse_const_dn_vec_descr_t dn_vec_descr) // non-const descriptor supported

此函数释放为稠密向量描述符 dn_vec_descr 分配的主机内存。

参数。

输入/输出

含义

dn_vec_descr

输入

稠密向量描述符

有关返回状态的描述,请参阅 nvpl_sparse_status_t


nvpl_sparse_dn_vec_get()#

nvpl_sparse_status_t
nvpl_sparse_dn_vec_get(nvpl_sparse_dn_vec_descr_t  dn_vec_descr,
                     int64_t*                      size,
                     void**                        values,
                     nvpl_sparse_data_type*        value_type)

nvpl_sparse_status_t
nvpl_sparse_const_dn_vec_get(nvpl_sparse_const_dn_vec_descr_t  dn_vec_descr,
                     int64_t*                                  size,
                     const void**                              values,
                     nvpl_sparse_data_type*                    value_type)

此函数返回稠密向量描述符 dn_vec_descr 的字段。

参数。

输入/输出

含义

dn_vec_descr

输入

稠密向量描述符

size

输出

稠密向量的大小

values

输出

稠密向量的值。包含 nnz 个元素的数组

value_type

输出

枚举器,指定 values 的数据类型

有关返回状态的描述,请参阅 nvpl_sparse_status_t


nvpl_sparse_dn_vec_get_values()#

nvpl_sparse_status_t
nvpl_sparse_dn_vec_get_values(nvpl_sparse_dn_vec_descr_t dn_vec_descr,
                              void**                     values)

nvpl_sparse_status_t
nvpl_sparse_const_dn_vec_get_values(nvpl_sparse_const_dn_vec_descr_t dn_vec_descr,
                                    const void**                     values)

此函数返回稠密向量描述符 dn_vec_descrvalues 字段。

参数。

输入/输出

含义

dn_vec_descr

输入

稠密向量描述符

values

输出

稠密向量的值

有关返回状态的描述,请参阅 nvpl_sparse_status_t


nvpl_sparse_dn_vec_set_values()#

nvpl_sparse_status_t
nvpl_sparse_dn_vec_set_values(nvpl_sparse_dn_vec_descr_t dn_vec_descr,
                              void*                      values)

此函数设置稠密向量描述符 dn_vec_descrvalues 字段。

参数。

输入/输出

含义

dn_vec_descr

输入

稠密向量描述符

values

输入

稠密向量的值。包含 size 个元素的数组

nvpl_sparse_dn_vec_set_values() 具有以下约束

  • values 必须与 dn_vec_descr 中指定的数据类型的大小对齐。有关数据类型的描述,请参阅 nvpl_Sparse_data_type_t

有关返回状态的描述,请参阅 nvpl_sparse_status_t