Implement cusparseXcoo2csr.

This commit is contained in:
Seunghoon Lee 2024-03-20 10:29:09 +09:00
parent cd1e0a3d50
commit f52edbd132
No known key found for this signature in database
GPG key ID: 91024D13C6CA4722

View file

@ -1014,6 +1014,25 @@ unsafe fn csr2cscex2(
))
}
unsafe fn xcoo2csr(
handle: *mut cusparseContext,
coo_row_ind: *const i32,
nnz: i32,
m: i32,
csr_sorted_row_ptr: *mut i32,
idx_base: cusparseIndexBase_t,
) -> cusparseStatus_t {
let idx_base = index_base(idx_base);
to_cuda(rocsparse_coo2csr(
handle.cast(),
coo_row_ind,
nnz,
m,
csr_sorted_row_ptr,
idx_base,
))
}
unsafe fn destory_dnvec(dn_vec_descr: *mut cusparseDnVecDescr) -> cusparseStatus_t {
if dn_vec_descr == ptr::null_mut() {
cusparseStatus_t::CUSPARSE_STATUS_SUCCESS