本文整理汇总了C++中DistSparseMatrix::LocalRow方法的典型用法代码示例。如果您正苦于以下问题：C++ DistSparseMatrix::LocalRow方法的具体用法？C++ DistSparseMatrix::LocalRow怎么用？C++ DistSparseMatrix::LocalRow使用的例子？那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类DistSparseMatrix的用法示例。
在下文中一共展示了DistSparseMatrix::LocalRow方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的C++代码示例。
示例1: Q
void IPM
( const DistSparseMatrix<Real>& A,
const DistMultiVec<Real>& d,
Real lambda,
DistMultiVec<Real>& x,
const qp::affine::Ctrl<Real>& ctrl )
{
EL_DEBUG_CSE
const Int m = A.Height();
const Int n = A.Width();
mpi::Comm comm = A.Comm();
DistSparseMatrix<Real> Q(comm), AHat(comm), G(comm);
DistMultiVec<Real> c(comm), b(comm), h(comm);
auto& dLoc = d.LockedMatrix();
auto& cLoc = c.Matrix();
auto& hLoc = h.Matrix();
// Q := | I 0 0 |
// | 0 0 0 |
// | 0 0 0 |
// ==============
Zeros( Q, n+m+1, n+m+1 );
{
// Count the number of local entries in the top-left I
// ---------------------------------------------------
Int numLocalUpdates = 0;
for( Int iLoc=0; iLoc<Q.LocalHeight(); ++iLoc )
if( Q.GlobalRow(iLoc) < n )
++numLocalUpdates;
else
break;
Q.Reserve( numLocalUpdates );
for( Int iLoc=0; iLoc<Q.LocalHeight(); ++iLoc )
if( Q.GlobalRow(iLoc) < n )
Q.QueueLocalUpdate( iLoc, Q.GlobalRow(iLoc), Real(1) );
Q.ProcessLocalQueues();
}
// c := [0;0;lambda]
// =================
Zeros( c, n+m+1, 1 );
for( Int iLoc=0; iLoc<c.LocalHeight(); ++iLoc )
if( c.GlobalRow(iLoc) > n )
cLoc(iLoc) = lambda;
// AHat = []
// =========
Zeros( AHat, 0, n+m+1 );
// b = []
// ======
Zeros( b, 0, 1 );
// G := |-diag(d) A, -d, -I|
// | 0, 0, -I|
// =========================
Zeros( G, 2*m, n+m+1 );
G.Reserve
( A.NumLocalEntries()+d.LocalHeight()+G.LocalHeight(),
A.NumLocalEntries()+d.LocalHeight() );
for( Int e=0; e<A.NumLocalEntries(); ++e )
{
const Int i = A.Row(e);
const Int j = A.Col(e);
const Int iLoc = A.LocalRow(i);
const Real value = -dLoc(iLoc)*A.Value(e);
G.QueueUpdate( i, j, value );
}
for( Int iLoc=0; iLoc<d.LocalHeight(); ++iLoc )
{
const Int i = d.GlobalRow(iLoc);
G.QueueUpdate( i, n, -dLoc(iLoc) );
}
for( Int iLoc=0; iLoc<G.LocalHeight(); ++iLoc )
{
const Int i = G.GlobalRow(iLoc);
if( i < m )
G.QueueLocalUpdate( iLoc, i+n+1, Real(-1) );
else
G.QueueLocalUpdate( iLoc, (i-m)+n+1, Real(-1) );
}
G.ProcessQueues();
// h := [-ones(m,1); zeros(m,1)]
// =============================
Zeros( h, 2*m, 1 );
for( Int iLoc=0; iLoc<h.LocalHeight(); ++iLoc )
if( h.GlobalRow(iLoc) < m )
hLoc(iLoc) = Real(-1);
else
break;
// Solve the affine QP
// ===================
DistMultiVec<Real> y(comm), z(comm), s(comm);
QP( Q, AHat, G, b, c, h, x, y, z, s, ctrl );
}