/dports/science/hypre/hypre-2.23.0/src/parcsr_ls/ |
H A D | par_interp_trunc_device.c | 122 HYPRE_Int nnz_diag = hypre_CSRMatrixNumNonzeros(P_diag); in hypre_BoomerAMGInterpTruncationDevice() local 124 HYPRE_Int nnz_P = nnz_diag + nnz_offd; in hypre_BoomerAMGInterpTruncationDevice() 143 hypreDevice_CsrRowPtrsToIndices_v2(nrows, nnz_diag, P_diag_i, P_i); in hypre_BoomerAMGInterpTruncationDevice() 144 hypreDevice_CsrRowPtrsToIndices_v2(nrows, nnz_offd, P_offd_i, P_i + nnz_diag); in hypre_BoomerAMGInterpTruncationDevice() 146 hypre_TMemcpy(P_j, P_diag_j, HYPRE_Int, nnz_diag, HYPRE_MEMORY_DEVICE, HYPRE_MEMORY_DEVICE); in hypre_BoomerAMGInterpTruncationDevice() 148 HYPRE_THRUST_CALL(transform, P_offd_j, P_offd_j + nnz_offd, P_j + nnz_diag, -_1 - 2); in hypre_BoomerAMGInterpTruncationDevice() 150 …hypre_TMemcpy(P_a, P_diag_a, HYPRE_Real, nnz_diag, HYPRE_MEMORY_DEVICE, HYPRE_MEMORY_DE… in hypre_BoomerAMGInterpTruncationDevice() 151 …hypre_TMemcpy(P_a + nnz_diag, P_offd_a, HYPRE_Real, nnz_offd, HYPRE_MEMORY_DEVICE, HYPRE_MEMORY_DE… in hypre_BoomerAMGInterpTruncationDevice() 166 if (nnz_diag) in hypre_BoomerAMGInterpTruncationDevice() 178 hypre_assert(new_nnz_diag <= nnz_diag); in hypre_BoomerAMGInterpTruncationDevice() [all …]
|
H A D | par_lr_restr_device.c | 49 HYPRE_Int nnz_diag, nnz_offd; in hypre_BoomerAMGBuildRestrNeumannAIRDevice() local 231 nnz_diag = nnz_diag_Z + n_cpts; in hypre_BoomerAMGBuildRestrNeumannAIRDevice() 236 R_diag_j = hypre_CTAlloc(HYPRE_Int, nnz_diag, HYPRE_MEMORY_DEVICE); in hypre_BoomerAMGBuildRestrNeumannAIRDevice() 237 R_diag_a = hypre_CTAlloc(HYPRE_Complex, nnz_diag, HYPRE_MEMORY_DEVICE); in hypre_BoomerAMGBuildRestrNeumannAIRDevice() 264 nnz_diag, in hypre_BoomerAMGBuildRestrNeumannAIRDevice()
|
H A D | par_lr_restr.c | 125 HYPRE_Int nnz_diag, nnz_offd, cnt_diag, cnt_offd; in hypre_BoomerAMGBuildRestrDist2AIR() local 928 nnz_diag = cnt_diag; in hypre_BoomerAMGBuildRestrDist2AIR() 933 R_diag_j = hypre_CTAlloc(HYPRE_Int, nnz_diag, HYPRE_MEMORY_HOST); in hypre_BoomerAMGBuildRestrDist2AIR() 934 R_diag_data = hypre_CTAlloc(HYPRE_Complex, nnz_diag, HYPRE_MEMORY_HOST); in hypre_BoomerAMGBuildRestrDist2AIR() 1496 hypre_assert(cnt_diag == nnz_diag); in hypre_BoomerAMGBuildRestrDist2AIR() 1563 nnz_diag, in hypre_BoomerAMGBuildRestrDist2AIR() 1698 HYPRE_Int nnz_diag, nnz_offd, cnt_diag, cnt_offd; in hypre_BoomerAMGBuildRestrNeumannAIRHost() local 1914 nnz_diag = hypre_CSRMatrixNumNonzeros(Z_diag) + n_cpts; in hypre_BoomerAMGBuildRestrNeumannAIRHost() 1919 R_diag_j = hypre_CTAlloc(HYPRE_Int, nnz_diag, HYPRE_MEMORY_DEVICE); in hypre_BoomerAMGBuildRestrNeumannAIRHost() 1969 hypre_assert(cnt_diag == nnz_diag); in hypre_BoomerAMGBuildRestrNeumannAIRHost() [all …]
|
H A D | par_mod_multi_interp.c | 549 HYPRE_Int nnz_diag, nnz_offd; in hypre_GenerateMultipassPi() local 682 nnz_diag = 0; in hypre_GenerateMultipassPi() 693 nnz_diag++; in hypre_GenerateMultipassPi() 713 P_diag_j = hypre_CTAlloc(HYPRE_Int, nnz_diag, HYPRE_MEMORY_HOST); in hypre_GenerateMultipassPi() 714 P_diag_data = hypre_CTAlloc(HYPRE_Real, nnz_diag, HYPRE_MEMORY_HOST); in hypre_GenerateMultipassPi() 874 HYPRE_Int nnz_diag, nnz_offd; in hypre_GenerateMultiPi() local 1009 nnz_diag = 0; in hypre_GenerateMultiPi() 1020 nnz_diag++; in hypre_GenerateMultiPi() 1040 Q_diag_j = hypre_CTAlloc(HYPRE_Int, nnz_diag, HYPRE_MEMORY_HOST); in hypre_GenerateMultiPi() 1041 Q_diag_data = hypre_CTAlloc(HYPRE_Real, nnz_diag, HYPRE_MEMORY_HOST); in hypre_GenerateMultiPi()
|
H A D | par_restr.c | 100 HYPRE_Int nnz_diag, nnz_offd, cnt_diag, cnt_offd; in hypre_BoomerAMGBuildRestrAIR() local 233 nnz_diag = cnt_diag; in hypre_BoomerAMGBuildRestrAIR() 238 R_diag_j = hypre_CTAlloc(HYPRE_Int, nnz_diag,HYPRE_MEMORY_HOST); in hypre_BoomerAMGBuildRestrAIR() 239 R_diag_data = hypre_CTAlloc(HYPRE_Complex, nnz_diag,HYPRE_MEMORY_HOST); in hypre_BoomerAMGBuildRestrAIR() 696 hypre_assert(cnt_diag == nnz_diag); in hypre_BoomerAMGBuildRestrAIR() 752 nnz_diag, in hypre_BoomerAMGBuildRestrAIR()
|
H A D | par_interp_device.c | 1048 HYPRE_Int nnz_diag, nnz_offd; in hypre_BoomerAMGBuildInterpOnePntDevice() local 1165 … hypre_TMemcpy(&nnz_diag, &P_diag_i[n_fine], HYPRE_Int, 1, HYPRE_MEMORY_HOST, HYPRE_MEMORY_DEVICE); in hypre_BoomerAMGBuildInterpOnePntDevice() 1168 P_diag_j = hypre_TAlloc(HYPRE_Int, nnz_diag, HYPRE_MEMORY_DEVICE); in hypre_BoomerAMGBuildInterpOnePntDevice() 1169 P_diag_data = hypre_TAlloc(HYPRE_Real, nnz_diag, HYPRE_MEMORY_DEVICE); in hypre_BoomerAMGBuildInterpOnePntDevice() 1178 nnz_diag, in hypre_BoomerAMGBuildInterpOnePntDevice() 1186 P_diag_j_temp_compressed = hypre_TAlloc(HYPRE_Int, nnz_diag, HYPRE_MEMORY_DEVICE); in hypre_BoomerAMGBuildInterpOnePntDevice() 1205 P_diag_j_temp_compressed + nnz_diag, in hypre_BoomerAMGBuildInterpOnePntDevice() 1272 nnz_diag, in hypre_BoomerAMGBuildInterpOnePntDevice()
|
H A D | par_mod_multi_interp_device.c | 746 HYPRE_Int nnz_diag, nnz_offd; in hypre_GenerateMultipassPiDevice() local 868 …hypre_TMemcpy(&nnz_diag, &P_diag_i[num_points], HYPRE_Int, 1, HYPRE_MEMORY_HOST, HYPRE_MEMORY_DEVI… in hypre_GenerateMultipassPiDevice() 873 P_diag_j = hypre_TAlloc(HYPRE_Int, nnz_diag, HYPRE_MEMORY_DEVICE); in hypre_GenerateMultipassPiDevice() 874 P_diag_data = hypre_TAlloc(HYPRE_Real, nnz_diag, HYPRE_MEMORY_DEVICE); in hypre_GenerateMultipassPiDevice() 916 nnz_diag, in hypre_GenerateMultipassPiDevice() 993 HYPRE_Int nnz_diag, nnz_offd; in hypre_GenerateMultiPiDevice() local 1116 …hypre_TMemcpy(&nnz_diag, &Q_diag_i[num_points], HYPRE_Int, 1, HYPRE_MEMORY_HOST, HYPRE_MEMORY_DEVI… in hypre_GenerateMultiPiDevice() 1121 Q_diag_j = hypre_TAlloc(HYPRE_Int, nnz_diag, HYPRE_MEMORY_DEVICE); in hypre_GenerateMultiPiDevice() 1122 Q_diag_data = hypre_TAlloc(HYPRE_Real, nnz_diag, HYPRE_MEMORY_DEVICE); in hypre_GenerateMultiPiDevice() 1168 nnz_diag, in hypre_GenerateMultiPiDevice()
|
H A D | par_strength_device.c | 395 HYPRE_Int nnz_diag = hypre_CSRMatrixNumNonzeros(A_diag); in hypre_BoomerAMGMakeSocFromSDevice() local 396 HYPRE_Int *soc_diag = hypre_TAlloc(HYPRE_Int, nnz_diag, HYPRE_MEMORY_DEVICE); in hypre_BoomerAMGMakeSocFromSDevice()
|
H A D | par_sv_interp.c | 265 HYPRE_Int nnz_diag, nnz_offd, fcn_num, num_elements; in hypre_BoomerAMG_GMExpandInterp() local 401 nnz_diag = P_diag_size; in hypre_BoomerAMG_GMExpandInterp() 484 new_nnz_diag = nnz_diag + nnz_diag*num_smooth_vecs; in hypre_BoomerAMG_GMExpandInterp()
|
H A D | par_sv_interp_ln.c | 113 HYPRE_Int nnz_diag, nnz_offd; in hypre_BoomerAMG_LNExpandInterp() local 278 nnz_diag = P_diag_size; in hypre_BoomerAMG_LNExpandInterp() 498 new_nnz_diag = nnz_diag + nnz_diag*num_smooth_vecs; in hypre_BoomerAMG_LNExpandInterp()
|
H A D | par_interp.c | 3824 HYPRE_Int nnz_diag, nnz_offd, cnt_diag, cnt_offd; in hypre_BoomerAMGBuildInterpOnePntHost() local 4033 nnz_diag = cnt_diag + n_cpts; in hypre_BoomerAMGBuildInterpOnePntHost() 4038 P_diag_j = hypre_CTAlloc(HYPRE_Int, nnz_diag,HYPRE_MEMORY_DEVICE); in hypre_BoomerAMGBuildInterpOnePntHost() 4039 P_diag_data = hypre_CTAlloc(HYPRE_Real, nnz_diag,HYPRE_MEMORY_DEVICE); in hypre_BoomerAMGBuildInterpOnePntHost() 4113 hypre_assert(cnt_diag == nnz_diag); in hypre_BoomerAMGBuildInterpOnePntHost() 4171 nnz_diag, in hypre_BoomerAMGBuildInterpOnePntHost()
|
H A D | ams.c | 2768 HYPRE_Int nnz_diag = hypre_CSRMatrixNumNonzeros(hypre_ParCSRMatrixDiag(B)); in hypre_AMSSetup() local 2773 hypre_assert(nnz == nnz_diag); in hypre_AMSSetup() 2777 if (nnz_diag) in hypre_AMSSetup() 2781 … thrust::make_transform_iterator(B_diag_data + nnz_diag, absolute_value<HYPRE_Real>()), in hypre_AMSSetup()
|
/dports/science/hypre/hypre-2.23.0/src/parcsr_mv/ |
H A D | par_csr_matop.c | 2582 nnz_diag = nnz11; in hypre_ParCSRMatrixExtractSubmatrices() 2633 nnz_diag = nnz12; in hypre_ParCSRMatrixExtractSubmatrices() 2671 if (nnz > nnz_diag) in hypre_ParCSRMatrixExtractSubmatrices() 2697 nnz_diag = nnz21; in hypre_ParCSRMatrixExtractSubmatrices() 2754 nnz_diag = nnz22; in hypre_ParCSRMatrixExtractSubmatrices() 2939 nnz_diag = nnz11; in hypre_ParCSRMatrixExtractRowSubmatrices() 3016 nnz_diag = nnz21; in hypre_ParCSRMatrixExtractRowSubmatrices() 6047 A_diag_j[nnz_diag] = col; in hypre_ParCSRMatrixDropSmallEntriesHost() 6048 A_diag_a[nnz_diag] = val; in hypre_ParCSRMatrixDropSmallEntriesHost() 6049 nnz_diag ++; in hypre_ParCSRMatrixDropSmallEntriesHost() [all …]
|
/dports/math/casadi/casadi-3.5.5/casadi/core/ |
H A D | generic_matrix.hpp | 88 casadi_int nnz_diag() const; 879 casadi_int GenericMatrix<MatType>::nnz_diag() const { in nnz_diag() function in casadi::GenericMatrix 880 return sparsity().nnz_diag(); in nnz_diag() 1079 casadi_assert(x.nnz_upper()-x.nnz_diag()==0, in tril2symm() 1103 casadi_assert(x.nnz_lower()-x.nnz_diag()==0, in triu2symm()
|
H A D | sparsity_internal.hpp | 413 casadi_int nnz_diag() const;
|
H A D | sparsity.hpp | 326 casadi_int nnz_diag() const;
|
H A D | sparsity.cpp | 340 casadi_int Sparsity::nnz_diag() const { in nnz_diag() function in casadi::Sparsity 341 return (*this)->nnz_diag(); in nnz_diag()
|
H A D | sparsity_internal.cpp | 1848 casadi_int SparsityInternal::nnz_diag() const { in nnz_diag() function in casadi::SparsityInternal
|
/dports/math/mfem/mfem-4.3/fem/ |
H A D | pfespace.cpp | 2411 HYPRE_Int nnz_diag = 0, nnz_offd = 0; in MakeVDimHypreMatrix() local 2421 nnz_diag += vdim; in MakeVDimHypreMatrix() 2447 HYPRE_Int *J_diag = Memory<HYPRE_Int>(nnz_diag); in MakeVDimHypreMatrix() 2450 double *A_diag = Memory<double>(nnz_diag); in MakeVDimHypreMatrix() 2458 nnz_diag = nnz_offd = 0; in MakeVDimHypreMatrix() 2466 I_diag[vrow] = nnz_diag; in MakeVDimHypreMatrix() 2475 J_diag[nnz_diag] = elem.column + vd*vdim_offset - first_col; in MakeVDimHypreMatrix() 2476 A_diag[nnz_diag++] = elem.value; in MakeVDimHypreMatrix() 2488 I_diag[vrow] = nnz_diag; in MakeVDimHypreMatrix()
|
/dports/math/casadi/casadi-3.5.5/test/python/ |
H A D | matrix.py | 668 self.assertEqual(Sparsity.diag(10).nnz_diag(),10) 673 self.assertEqual(Sparsity.dense(10,10).nnz_diag(),10) 675 self.assertEqual(sparsify(DM([[1,1,0],[1,0,1],[0,0,0]])).nnz_diag(),1) 687 print(DM.ones(5,5).nnz_upper()-DM.ones(5,5).nnz_diag())
|
/dports/science/hypre/hypre-2.23.0/src/IJ_mv/ |
H A D | IJMatrix_parcsr_device.c | 746 HYPRE_Int nnz_diag = hypre_CSRMatrixNumNonzeros(diag); in hypre_IJMatrixSetConstantValuesParCSRDevice() local 749 HYPRE_THRUST_CALL( fill_n, diag_data, nnz_diag, value ); in hypre_IJMatrixSetConstantValuesParCSRDevice()
|
H A D | IJMatrix_parcsr.c | 1025 HYPRE_Int nnz_diag = hypre_CSRMatrixNumNonzeros(diag); in hypre_IJMatrixSetConstantValuesParCSRHost() local 1032 for (ii = 0; ii < nnz_diag; ii++) in hypre_IJMatrixSetConstantValuesParCSRHost()
|
/dports/math/casadi/casadi-3.5.5/swig/ |
H A D | doc.i | 32271 %feature("docstring") casadi::GenericMatrixCommon::nnz_diag() const " 53250 %feature("docstring") casadi::GenericMatrixCommon::nnz_diag() const " 56012 %feature("docstring") casadi::GenericMatrixCommon::nnz_diag() const " 80799 %feature("docstring") casadi::Sparsity::nnz_diag() const "
|
H A D | doc_merged.i | 42480 %feature("docstring") casadi::MatrixCommon::nnz_diag " 43966 %feature("docstring") casadi::MX::nnz_diag "
|