Commit cf4a141e authored by Jakub Klinkovský's avatar Jakub Klinkovský
Browse files

Fixed bug in Ellpack for rectangular matrices

parent 8dea5268

src/TNL/Matrices/Ellpack_impl.h

+4 −2
Original line number Original line Diff line number Diff line
@@ -75,7 +75,7 @@ void Ellpack< Real, Device, Index >::setDimensions( const IndexType rows, 
   this->rows = rows;

   this->rows = rows;

   this->columns = columns;

   this->columns = columns;

   if( std::is_same< Device, Devices::Cuda >::value )

   if( std::is_same< Device, Devices::Cuda >::value )

      this->alignedRows = roundToMultiple( columns, Devices::Cuda::getWarpSize() );

      this->alignedRows = roundToMultiple( rows, Devices::Cuda::getWarpSize() );

   else this->alignedRows = rows;

   else this->alignedRows = rows;

   if( this->rowLengths != 0 )

   if( this->rowLengths != 0 )

      allocateElements();

      allocateElements();
@@ -142,7 +142,9 @@ void Ellpack< Real, Device, Index >::setLike( const Ellpack< Real2, Device2, Ind 
{

{

   Sparse< Real, Device, Index >::setLike( matrix );

   Sparse< Real, Device, Index >::setLike( matrix );

   this->rowLengths = matrix.rowLengths;

   this->rowLengths = matrix.rowLengths;

   this->alignedRows = matrix.alignedRows;

   if( std::is_same< Device, Devices::Cuda >::value )

      this->alignedRows = roundToMultiple( this->getRows(), Devices::Cuda::getWarpSize() );

   else this->alignedRows = this->getRows();

}

}





template< typename Real,

template< typename Real,