本文整理汇总了C++中DistSparseMatrix::ProcessQueues方法的典型用法代码示例。如果您正苦于以下问题:C++ DistSparseMatrix::ProcessQueues方法的具体用法?C++ DistSparseMatrix::ProcessQueues怎么用?C++ DistSparseMatrix::ProcessQueues使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类DistSparseMatrix
的用法示例。
在下文中一共展示了DistSparseMatrix::ProcessQueues方法的3个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的C++代码示例。
示例1: Helmholtz
void Helmholtz( DistSparseMatrix<F>& H, Int nx, Int ny, F shift )
{
DEBUG_CSE
typedef Base<F> Real;
const Int n = nx*ny;
Zeros( H, n, n );
const Real hxInv = nx+1;
const Real hyInv = ny+1;
const Real hxInvSquared = hxInv*hxInv;
const Real hyInvSquared = hyInv*hyInv;
const F mainTerm = 2*(hxInvSquared+hyInvSquared) - shift;
const Int localHeight = H.LocalHeight();
H.Reserve( 5*localHeight );
for( Int iLoc=0; iLoc<localHeight; ++iLoc )
{
const Int i = H.GlobalRow(iLoc);
const Int x = i % nx;
const Int y = i/nx;
H.QueueUpdate( i, i, mainTerm );
if( x != 0 )
H.QueueUpdate( i, i-1, -hxInvSquared );
if( x != nx-1 )
H.QueueUpdate( i, i+1, -hxInvSquared );
if( y != 0 )
H.QueueUpdate( i, i-nx, -hyInvSquared );
if( y != ny-1 )
H.QueueUpdate( i, i+nx, -hyInvSquared );
}
H.ProcessQueues();
}
示例2: func
void EntrywiseMap
( const DistSparseMatrix<S>& A,
DistSparseMatrix<T>& B,
function<T(S)> func )
{
DEBUG_CSE
const Int numEntries = A.vals_.size();
const Int numRemoteEntries = A.remoteVals_.size();
B.distGraph_ = A.distGraph_;
B.vals_.resize( numEntries );
for( Int k=0; k<numEntries; ++k )
B.vals_[k] = func(A.vals_[k]);
B.remoteVals_.resize( numRemoteEntries );
for( Int k=0; k<numRemoteEntries; ++k )
B.remoteVals_[k] = func(A.remoteVals_[k]);
B.ProcessQueues();
}
示例3: JordanCholesky
void JordanCholesky( DistSparseMatrix<T>& A, Int n )
{
DEBUG_ONLY(CSE cse("JordanCholesky"))
Zeros( A, n, n );
const Int localHeight = A.LocalHeight();
A.Reserve( 3*localHeight );
for( Int iLoc=0; iLoc<localHeight; ++iLoc )
{
const Int i = A.GlobalRow(iLoc);
if( i == 0 )
A.QueueUpdate( i, i, T(1) );
else
A.QueueUpdate( i, i, T(5) );
if( i > 0 )
A.QueueUpdate( i, i-1, T(2) );
if( i < n-1 )
A.QueueUpdate( i, i+1, T(2) );
}
A.ProcessQueues();
}