Commit e945cfb9 authored by Tomáš Oberhuber's avatar Tomáš Oberhuber
Browse files

Added BiEllpack to spmv-legacy benchmark.

parent e9f25336

src/Benchmarks/SpMV/spmv-legacy.h

+8 −0
Original line number Diff line number Diff line
@@ -34,6 +34,7 @@ 
#include <TNL/Containers/Segments/Ellpack.h>

#include <TNL/Containers/Segments/SlicedEllpack.h>

#include <TNL/Containers/Segments/ChunkedEllpack.h>

#include <TNL/Containers/Segments/BiEllpack.h>

using namespace TNL::Matrices;



#include "cusparseCSRMatrix.h"
@@ -68,6 +69,12 @@ using ChunkedEllpackSegments = Containers::Segments::ChunkedEllpack< Device, Ind 
template< typename Real, typename Device, typename Index >

using SparseMatrix_ChunkedEllpack = Matrices::SparseMatrix< Real, Device, Index, Matrices::GeneralMatrix, ChunkedEllpackSegments >;



template< typename Device, typename Index, typename IndexAllocator >

using BiEllpackSegments = Containers::Segments::BiEllpack< Device, Index, IndexAllocator >;



template< typename Real, typename Device, typename Index >

using SparseMatrix_BiEllpack = Matrices::SparseMatrix< Real, Device, Index, Matrices::GeneralMatrix, BiEllpackSegments >;



// Legacy formats

template< typename Real, typename Device, typename Index >

using SparseMatrixLegacy_CSR_Scalar = Matrices::Legacy::CSR< Real, Device, Index, Matrices::Legacy::CSRScalar >;
@@ -295,6 +302,7 @@ benchmarkSpmvSynthetic( Benchmark& benchmark, 
   benchmarkSpMV< Real, Matrices::Legacy::ChunkedEllpack >( benchmark, hostOutVector, inputFileName, verboseMR );

   benchmarkSpMV< Real, SparseMatrix_ChunkedEllpack      >( benchmark, hostOutVector, inputFileName, verboseMR );

   benchmarkSpMV< Real, Matrices::Legacy::BiEllpack      >( benchmark, hostOutVector, inputFileName, verboseMR );

   benchmarkSpMV< Real, SparseMatrix_BiEllpack           >( benchmark, hostOutVector, inputFileName, verboseMR );

   /* AdEllpack is broken

   benchmarkSpMV< Real, Matrices::AdEllpack              >( benchmark, hostOutVector, inputFileName, verboseMR );

    */