Skip to content

Commit f52edbd

Browse files
committed
Implement cusparseXcoo2csr.
1 parent cd1e0a3 commit f52edbd

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

zluda_sparse/src/lib.rs

+19
Original file line numberDiff line numberDiff line change
@@ -1014,6 +1014,25 @@ unsafe fn csr2cscex2(
10141014
))
10151015
}
10161016

1017+
unsafe fn xcoo2csr(
1018+
handle: *mut cusparseContext,
1019+
coo_row_ind: *const i32,
1020+
nnz: i32,
1021+
m: i32,
1022+
csr_sorted_row_ptr: *mut i32,
1023+
idx_base: cusparseIndexBase_t,
1024+
) -> cusparseStatus_t {
1025+
let idx_base = index_base(idx_base);
1026+
to_cuda(rocsparse_coo2csr(
1027+
handle.cast(),
1028+
coo_row_ind,
1029+
nnz,
1030+
m,
1031+
csr_sorted_row_ptr,
1032+
idx_base,
1033+
))
1034+
}
1035+
10171036
unsafe fn destory_dnvec(dn_vec_descr: *mut cusparseDnVecDescr) -> cusparseStatus_t {
10181037
if dn_vec_descr == ptr::null_mut() {
10191038
cusparseStatus_t::CUSPARSE_STATUS_SUCCESS

0 commit comments

Comments
 (0)