//------------------------------------------------------------------------------ // GB_aop: assign/subassign kernels with accum //------------------------------------------------------------------------------ // SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2023, All Rights Reserved. // SPDX-License-Identifier: Apache-2.0 //------------------------------------------------------------------------------ // C(I,J) += A #include "GB.h" #include "GB_control.h" #include "GB_ek_slice.h" #include "GB_aop__include.h" // accum operator #define GB_ACCUM_OP(z,x,y) z = GB_FC32_iseq (x, y) #define GB_Z_TYPE GxB_FC32_t #define GB_X_TYPE GxB_FC32_t #define GB_Y_TYPE GxB_FC32_t #define GB_DECLAREY(ywork) GxB_FC32_t ywork #define GB_COPY_aij_to_ywork(ywork,Ax,pA,A_iso) ywork = Ax [(A_iso) ? 0 : (pA)] // A and C matrices #define GB_A_TYPE GxB_FC32_t #define GB_C_TYPE GxB_FC32_t #define GB_DECLAREC(cwork) GxB_FC32_t cwork #define GB_COPY_aij_to_cwork(cwork,Ax,pA,A_iso) cwork = Ax [A_iso ? 0 : (pA)] #define GB_COPY_aij_to_C(Cx,pC,Ax,pA,A_iso,cwork) Cx [pC] = (A_iso) ? cwork : Ax [pA] #define GB_COPY_scalar_to_C(Cx,pC,cwork) Cx [pC] = cwork #define GB_AX_MASK(Ax,pA,asize) GB_MCAST (Ax, pA, sizeof (GxB_FC32_t)) // C(i,j) += ywork #define GB_ACCUMULATE_scalar(Cx,pC,ywork) \ GB_ACCUM_OP (Cx [pC], Cx [pC], ywork) // C(i,j) += (ytype) A(i,j) #define GB_ACCUMULATE_aij(Cx,pC,Ax,pA,A_iso,ywork) \ { \ if (A_iso) \ { \ GB_ACCUMULATE_scalar (Cx, pC, ywork) ; \ } \ else \ { \ /* A and Y have the same type here */ \ GB_ACCUMULATE_scalar (Cx, pC, Ax [pA]) ; \ } \ } // disable this operator and use the generic case if these conditions hold #if (defined(GxB_NO_ISEQ) || defined(GxB_NO_FC32) || defined(GxB_NO_ISEQ_FC32)) #define GB_DISABLE 1 #else #define GB_DISABLE 0 #endif #include "GB_assign_shared_definitions.h" //------------------------------------------------------------------------------ // C += A, accumulate a sparse matrix into a dense matrix //------------------------------------------------------------------------------ GrB_Info GB (_subassign_23__iseq_fc32) ( GrB_Matrix C, const GrB_Matrix A, GB_Werk Werk ) { #if GB_DISABLE return (GrB_NO_VALUE) ; #else { int nthreads_max = GB_Context_nthreads_max ( ) ; double chunk = GB_Context_chunk ( ) ; #include "GB_subassign_23_template.c" } return (GrB_SUCCESS) ; #endif } //------------------------------------------------------------------------------ // C += y, accumulate a scalar into a dense matrix //------------------------------------------------------------------------------ GrB_Info GB (_subassign_22__iseq_fc32) ( GrB_Matrix C, const GB_void *ywork_handle ) { #if GB_DISABLE return (GrB_NO_VALUE) ; #else { // get the scalar ywork for C += ywork, of type GB_Y_TYPE GB_Y_TYPE ywork = (*((GB_Y_TYPE *) ywork_handle)) ; int nthreads_max = GB_Context_nthreads_max ( ) ; double chunk = GB_Context_chunk ( ) ; #include "GB_subassign_22_template.c" return (GrB_SUCCESS) ; } return (GrB_SUCCESS) ; #endif }