add reduce operations

Author: abhinavmehndiratta

src/Operations/Reduce.jl

@@ -0,0 +1,83 @@
+function GrB_Matrix_reduce_Monoid(          # w<mask> = accum (w,reduce(A))
+    w::GrB_Vector,                          # input/output vector for results
+    mask::T,                                # optional mask for w, unused if NULL
+    accum::U,                               # optional accum for z=accum(w,t)
+    monoid::GrB_Monoid,                     # reduce operator for t=reduce(A)
+    A::GrB_Matrix,                          # first input:  matrix A
+    desc::V                                 # descriptor for w, mask, and A
+) where {T <: valid_vector_mask_types, U <: valid_accum_types, V <: valid_desc_types}
+
+    return GrB_Info(
+                ccall(
+                        dlsym(graphblas_lib, "GrB_Matrix_reduce_Monoid"),
+                        Cint,
+                        (Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}),
+                        w.p, mask.p, accum.p, monoid.p, A.p, desc.p
+                    )
+                )
+end
+
+function GrB_Matrix_reduce_BinaryOp(        # w<mask> = accum (w,reduce(A))
+    w::GrB_Vector,                          # input/output vector for results
+    mask::T,                                # optional mask for w, unused if NULL
+    accum::U,                               # optional accum for z=accum(w,t)
+    op::GrB_BinaryOp,                       # reduce operator for t=reduce(A)
+    A::GrB_Matrix,                          # first input:  matrix A
+    desc::V                                 # descriptor for w, mask, and A
+) where {T <: valid_vector_mask_types, U <: valid_accum_types, V <: valid_desc_types}
+
+    return GrB_Info(
+                ccall(
+                        dlsym(graphblas_lib, "GrB_Matrix_reduce_BinaryOp"),
+                        Cint,
+                        (Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}),
+                        w.p, mask.p, accum.p, op.p, A.p, desc.p
+                    )
+                )
+end
+
+function GrB_Vector_reduce(     # c = accum (c, reduce_to_scalar (u))
+    accum::U,                   # optional accum for c=accum(c,t)
+    monoid::GrB_Monoid,         # monoid to do the reduction
+    u::GrB_Vector{T},           # vector to reduce
+    desc::V                     # descriptor (currently unused)
+) where {T <: valid_types, U <: valid_accum_types, V <: valid_desc_types}
+
+    scalar = Ref(T(0))
+    fn_name = "GrB_Vector_reduce_" * suffix(T)
+
+    res =   GrB_Info(
+                ccall(
+                        dlsym(graphblas_lib, fn_name),
+                        Cint,
+                        (Ptr{T}, Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}),
+                        scalar, accum.p, monoid.p, u.p, desc.p
+                    )
+                )
+
+    res != GrB_SUCCESS && return res
+    return scalar[]
+end
+
+function GrB_Matrix_reduce(     # c = accum (c, reduce_to_scalar (u))
+    accum::U,                   # optional accum for c=accum(c,t)
+    monoid::GrB_Monoid,         # monoid to do the reduction
+    A::GrB_Matrix{T},           # vector to reduce
+    desc::V                     # descriptor (currently unused)
+) where {T <: valid_types, U <: valid_accum_types, V <: valid_desc_types}
+
+    scalar = Ref(T(0))
+    fn_name = "GrB_Matrix_reduce_" * suffix(T)
+
+    res =   GrB_Info(
+                ccall(
+                        dlsym(graphblas_lib, fn_name),
+                        Cint,
+                        (Ptr{T}, Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}),
+                        scalar, A.p, monoid.p, u.p, desc.p
+                    )
+                )
+
+    res != GrB_SUCCESS && return res
+    return scalar[]
+end

src/SuiteSparseGraphBLAS.jl

@@ -147,6 +147,7 @@ include("Operations/Element_wise_multiplication.jl")
 include("Operations/Element_wise_addition.jl")
 include("Operations/Extract.jl")
 include("Operations/Apply.jl")
+include("Operations/Reduce.jl")
 
 export
 # Context Methods
@@ -192,6 +193,10 @@ GrB_Vector_extract, GrB_Matrix_extract, GrB_Col_extract,
 # Apply
 GrB_apply, GrB_Vector_apply, GrB_Matrix_apply
 
+# Reduce
+GrB_Matrix_reduce_Monoid, GrB_Matrix_reduce_BinaryOp, GrB_Matrix_reduce,
+GrB_Vector_reduce
+
 # Export global variables
 
 # Types