|
| | SpParMat3D (int nlayers) |
| |
| | SpParMat3D (const SpParMat< IT, NT, DER > &A2D, int nlayers, bool colsplit, bool special=false) |
| |
| | SpParMat3D (DER *myseq, std::shared_ptr< CommGrid3D > grid3d, bool colsplit, bool special=false) |
| |
| | SpParMat3D (const SpParMat3D< IT, NT, DER > &A3D, bool colsplit) |
| |
| | ~SpParMat3D () |
| |
| SpParMat< IT, NT, DER > | Convert2D () |
| |
| float | LoadImbalance () const |
| |
| void | FreeMemory () |
| |
| void | PrintInfo () const |
| |
| IT | getnrow () const |
| |
| IT | getncol () const |
| |
| IT | getnnz () const |
| |
| std::shared_ptr< SpParMat< IT, NT, DER > > | GetLayerMat () |
| |
| DER * | seqptr () const |
| |
| bool | isSpecial () const |
| |
| bool | isColSplit () const |
| |
| template<typename LIT > |
| int | Owner (IT total_m, IT total_n, IT grow, IT gcol, LIT &lrow, LIT &lcol) const |
| |
| void | LocalDim (IT total_m, IT total_n, IT &localm, IT &localn) const |
| |
| void | CalculateColSplitDistributionOfLayer (vector< typename DER::LocalIT > &divisions3d) |
| |
| bool | CheckSpParMatCompatibility () |
| |
| std::shared_ptr< CommGrid3D > | getcommgrid () const |
| |
| std::shared_ptr< CommGrid3D > | getcommgrid3D () const |
| |
| | SpParMat3D (int nlayers) |
| |
| | SpParMat3D (const SpParMat< IT, NT, DER > &A2D, int nlayers, bool colsplit, bool special=false) |
| |
| | SpParMat3D (DER *myseq, std::shared_ptr< CommGrid3D > grid3d, bool colsplit, bool special=false) |
| |
| | SpParMat3D (const SpParMat3D< IT, NT, DER > &A3D, bool colsplit) |
| |
| | ~SpParMat3D () |
| |
| SpParMat< IT, NT, DER > | Convert2D () |
| |
| float | LoadImbalance () const |
| |
| void | FreeMemory () |
| |
| void | PrintInfo () const |
| |
| IT | getnrow () const |
| |
| IT | getncol () const |
| |
| IT | getnnz () const |
| |
| std::shared_ptr< SpParMat< IT, NT, DER > > | GetLayerMat () |
| |
| DER * | seqptr () const |
| |
| bool | isSpecial () const |
| |
| bool | isColSplit () const |
| |
| template<typename LIT > |
| int | Owner (IT total_m, IT total_n, IT grow, IT gcol, LIT &lrow, LIT &lcol) const |
| |
| void | LocalDim (IT total_m, IT total_n, IT &localm, IT &localn) const |
| |
| void | CalculateColSplitDistributionOfLayer (vector< typename DER::LocalIT > &divisions3d) |
| |
| bool | CheckSpParMatCompatibility () |
| |
| std::shared_ptr< CommGrid3D > | getcommgrid () const |
| |
| std::shared_ptr< CommGrid3D > | getcommgrid3D () const |
| |
| | SpParMat3D (int nlayers) |
| |
| | SpParMat3D (const SpParMat< IT, NT, DER > &A2D, int nlayers, bool colsplit, bool special=false) |
| |
| | SpParMat3D (DER *myseq, std::shared_ptr< CommGrid3D > grid3d, bool colsplit, bool special=false) |
| |
| | SpParMat3D (const SpParMat3D< IT, NT, DER > &A3D, bool colsplit) |
| |
| | ~SpParMat3D () |
| |
| SpParMat< IT, NT, DER > | Convert2D () |
| |
| float | LoadImbalance () const |
| |
| void | FreeMemory () |
| |
| void | PrintInfo () const |
| |
| IT | getnrow () const |
| |
| IT | getncol () const |
| |
| IT | getnnz () const |
| |
| std::shared_ptr< SpParMat< IT, NT, DER > > | GetLayerMat () |
| |
| DER * | seqptr () const |
| |
| bool | isSpecial () const |
| |
| bool | isColSplit () const |
| |
| template<typename LIT > |
| int | Owner (IT total_m, IT total_n, IT grow, IT gcol, LIT &lrow, LIT &lcol) const |
| |
| void | LocalDim (IT total_m, IT total_n, IT &localm, IT &localn) const |
| |
| void | CalculateColSplitDistributionOfLayer (vector< typename DER::LocalIT > &divisions3d) |
| |
| bool | CheckSpParMatCompatibility () |
| |
| std::shared_ptr< CommGrid3D > | getcommgrid () const |
| |
| std::shared_ptr< CommGrid3D > | getcommgrid3D () const |
| |
|
| template<typename SR , typename NUO , typename UDERO , typename IU , typename NU1 , typename NU2 , typename UDER1 , typename UDER2 > |
| SpParMat3D< IU, NUO, UDERO > | Mult_AnXBn_SUMMA3D (SpParMat3D< IU, NU1, UDER1 > &A, SpParMat3D< IU, NU2, UDER2 > &B) |
| |
| template<typename SR , typename NUO , typename UDERO , typename IU , typename NU1 , typename NU2 , typename UDER1 , typename UDER2 > |
| SpParMat3D< IU, NUO, UDERO > | MemEfficientSpGEMM3D (SpParMat3D< IU, NU1, UDER1 > &A, SpParMat3D< IU, NU2, UDER2 > &B, int phases, NUO hardThreshold, IU selectNum, IU recoverNum, NUO recoverPct, int kselectVersion, int computationKernel, int64_t perProcessMemory) |
| |
| template<typename SR , typename NUO , typename UDERO , typename IU , typename NU1 , typename NU2 , typename UDER1 , typename UDER2 > |
| SpParMat3D< IU, NUO, UDERO > | Mult_AnXBn_SUMMA3D (SpParMat3D< IU, NU1, UDER1 > &A, SpParMat3D< IU, NU2, UDER2 > &B) |
| |
| template<typename SR , typename NUO , typename UDERO , typename IU , typename NU1 , typename NU2 , typename UDER1 , typename UDER2 > |
| SpParMat3D< IU, NUO, UDERO > | MemEfficientSpGEMM3D (SpParMat3D< IU, NU1, UDER1 > &A, SpParMat3D< IU, NU2, UDER2 > &B, int phases, NUO hardThreshold, IU selectNum, IU recoverNum, NUO recoverPct, int kselectVersion, int computationKernel, int64_t perProcessMemory) |
| |
| template<typename SR , typename NUO , typename UDERO , typename IU , typename NU1 , typename NU2 , typename UDER1 , typename UDER2 > |
| SpParMat3D< IU, NUO, UDERO > | Mult_AnXBn_SUMMA3D (SpParMat3D< IU, NU1, UDER1 > &A, SpParMat3D< IU, NU2, UDER2 > &B) |
| |
| template<typename SR , typename NUO , typename UDERO , typename IU , typename NU1 , typename NU2 , typename UDER1 , typename UDER2 > |
| SpParMat3D< IU, NUO, UDERO > | MemEfficientSpGEMM3D (SpParMat3D< IU, NU1, UDER1 > &A, SpParMat3D< IU, NU2, UDER2 > &B, int phases, NUO hardThreshold, IU selectNum, IU recoverNum, NUO recoverPct, int kselectVersion, int computationKernel, int64_t perProcessMemory) |
| |
Definition at line 35 of file SpParMat3D.h.