本文整理汇总了C++中DistSparseMatrix::LockedTargetBuffer方法的典型用法代码示例。如果您正苦于以下问题：C++ DistSparseMatrix::LockedTargetBuffer方法的具体用法？C++ DistSparseMatrix::LockedTargetBuffer怎么用？C++ DistSparseMatrix::LockedTargetBuffer使用的例子？那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类DistSparseMatrix的用法示例。
在下文中一共展示了DistSparseMatrix::LockedTargetBuffer方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的C++代码示例。
示例1: LogicError
void GetMappedDiagonal
( const DistSparseMatrix<T>& A,
DistMultiVec<S>& d,
function<S(const T&)> func,
Int offset )
{
EL_DEBUG_CSE
const Int m = A.Height();
const Int n = A.Width();
const T* valBuf = A.LockedValueBuffer();
const Int* colBuf = A.LockedTargetBuffer();
if( m != n )
LogicError("DistSparseMatrix GetMappedDiagonal assumes square matrix");
if( offset != 0 )
LogicError("DistSparseMatrix GetMappedDiagonal assumes offset=0");
d.SetGrid( A.Grid() );
d.Resize( El::DiagonalLength(m,n,offset), 1 );
Fill( d, S(1) );
S* dBuf = d.Matrix().Buffer();
const Int dLocalHeight = d.LocalHeight();
for( Int iLoc=0; iLoc<dLocalHeight; ++iLoc )
{
const Int i = d.GlobalRow(iLoc);
const Int thisOff = A.RowOffset(iLoc);
const Int nextOff = A.RowOffset(iLoc+1);
auto it = std::lower_bound( colBuf+thisOff, colBuf+nextOff, i );
if( *it == i )
{
const Int e = it-colBuf;
dBuf[iLoc] = func(valBuf[e]);
}
else
dBuf[iLoc] = func(0);
}
}