Home
last modified time | relevance | path

Searched refs:nnz_diag (Results 1 – 24 of 24) sorted by relevance

/dports/science/hypre/hypre-2.23.0/src/parcsr_ls/
H A Dpar_interp_trunc_device.c122 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 Dpar_lr_restr_device.c49 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 Dpar_lr_restr.c125 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 Dpar_mod_multi_interp.c549 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 Dpar_restr.c100 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 Dpar_interp_device.c1048 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 Dpar_mod_multi_interp_device.c746 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 Dpar_strength_device.c395 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 Dpar_sv_interp.c265 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 Dpar_sv_interp_ln.c113 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 Dpar_interp.c3824 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 Dams.c2768 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 Dpar_csr_matop.c2582 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 Dgeneric_matrix.hpp88 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 Dsparsity_internal.hpp413 casadi_int nnz_diag() const;
H A Dsparsity.hpp326 casadi_int nnz_diag() const;
H A Dsparsity.cpp340 casadi_int Sparsity::nnz_diag() const { in nnz_diag() function in casadi::Sparsity
341 return (*this)->nnz_diag(); in nnz_diag()
H A Dsparsity_internal.cpp1848 casadi_int SparsityInternal::nnz_diag() const { in nnz_diag() function in casadi::SparsityInternal
/dports/math/mfem/mfem-4.3/fem/
H A Dpfespace.cpp2411 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 Dmatrix.py668 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 DIJMatrix_parcsr_device.c746 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 DIJMatrix_parcsr.c1025 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 Ddoc.i32271 %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 Ddoc_merged.i42480 %feature("docstring") casadi::MatrixCommon::nnz_diag "
43966 %feature("docstring") casadi::MX::nnz_diag "