Skip to content

Commit eb0e2be

Browse files
add error checking in dropzeros!
1 parent 46eb87d commit eb0e2be

File tree

2 files changed

+16
-4
lines changed

2 files changed

+16
-4
lines changed

src/Interface/Object_Methods/Matrix_Methods.jl

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -448,6 +448,12 @@ Remove all zero entries from GraphBLAS matrix.
448448
"""
449449
function dropzeros!(A::GrB_Matrix)
450450
outp_replace_desc = GrB_Descriptor(Dict(GrB_OUTP => GrB_REPLACE))
451-
GrB_assign(A, A, GrB_NULL, A, GrB_ALL, 0, GrB_ALL, 0, outp_replace_desc)
452-
GrB_free(outp_replace_desc)
451+
res = GrB_assign(A, A, GrB_NULL, A, GrB_ALL, 0, GrB_ALL, 0, outp_replace_desc)
452+
if res != GrB_SUCCESS
453+
error(res)
454+
end
455+
res = GrB_free(outp_replace_desc)
456+
if res != GrB_SUCCESS
457+
error(res)
458+
end
453459
end

src/Interface/Object_Methods/Vector_Methods.jl

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -347,6 +347,12 @@ Remove all zero entries from GraphBLAS vector.
347347
"""
348348
function dropzeros!(v::GrB_Vector)
349349
outp_replace_desc = GrB_Descriptor(Dict(GrB_OUTP => GrB_REPLACE))
350-
GrB_assign(v, v, GrB_NULL, v, GrB_ALL, 0, outp_replace_desc)
351-
GrB_free(outp_replace_desc)
350+
res = GrB_assign(v, v, GrB_NULL, v, GrB_ALL, 0, outp_replace_desc)
351+
if res != GrB_SUCCESS
352+
error(res)
353+
end
354+
res = GrB_free(outp_replace_desc)
355+
if res != GrB_SUCCESS
356+
error(res)
357+
end
352358
end

0 commit comments

Comments
 (0)