From 2a40552495d234d1a171bbad26618ec80d84a064 Mon Sep 17 00:00:00 2001
From: Lukas Cejka <lukas.ostatek@gmail.com>
Date: Sat, 9 Mar 2019 18:37:57 +0100
Subject: [PATCH] Uncommented ifdef that served as a barrier for use of file
functions.
---
src/Benchmarks/SpMV/cusparseCSRMatrix.h | 158 ++++++++++++++++++++++++
1 file changed, 158 insertions(+)
create mode 100644 src/Benchmarks/SpMV/cusparseCSRMatrix.h
diff --git a/src/Benchmarks/SpMV/cusparseCSRMatrix.h b/src/Benchmarks/SpMV/cusparseCSRMatrix.h
new file mode 100644
index 0000000000..8ed210d9ad
--- /dev/null
+++ b/src/Benchmarks/SpMV/cusparseCSRMatrix.h
@@ -0,0 +1,158 @@
+/***************************************************************************
+ tnlCusparseCSR.h - description
+ -------------------
+ begin : Jul 3, 2014
+ copyright : (C) 2014 by Tomas Oberhuber
+ email : tomas.oberhuber@fjfi.cvut.cz
+ ***************************************************************************/
+
+/* See Copyright Notice in tnl/Copyright */
+
+#include <TNL/Assert.h>
+#include <TNL/Devices/Cuda.h>
+#ifdef HAVE_CUDA
+#include <cusparse.h>
+#endif
+
+namespace TNL {
+
+template< typename Real >
+class CusparseCSRBase
+{
+ public:
+ typedef Real RealType;
+ typedef Devices::Cuda DeviceType;
+ typedef Matrices::CSR< RealType, Devices::Cuda, int > MatrixType;
+
+ CusparseCSRBase()
+ : matrix( 0 )
+ {
+ };
+
+#ifdef HAVE_CUDA
+ void init( const MatrixType& matrix,
+ cusparseHandle_t* cusparseHandle )
+ {
+ this->matrix = &matrix;
+ this->cusparseHandle = cusparseHandle;
+ cusparseCreateMatDescr( & this->matrixDescriptor );
+ };
+#endif
+
+ int getRows() const
+ {
+ return matrix->getRows();
+ }
+
+ int getColumns() const
+ {
+ return matrix->getColumns();
+ }
+
+ int getNumberOfMatrixElements() const
+ {
+ return matrix->getNumberOfMatrixElements();
+ }
+
+
+ template< typename InVector,
+ typename OutVector >
+ void vectorProduct( const InVector& inVector,
+ OutVector& outVector ) const
+ {
+ TNL_ASSERT_TRUE( matrix, "matrix was not initialized" );
+#ifdef HAVE_CUDA
+ cusparseDcsrmv( *( this->cusparseHandle ),
+ CUSPARSE_OPERATION_NON_TRANSPOSE,
+ this->matrix->getRows(),
+ this->matrix->getColumns(),
+ this->matrix->values.getSize(),
+ 1.0,
+ this->matrixDescriptor,
+ this->matrix->values.getData(),
+ this->matrix->rowPointers.getData(),
+ this->matrix->columnIndexes.getData(),
+ inVector.getData(),
+ 1.0,
+ outVector.getData() );
+#endif
+ }
+
+ protected:
+
+ const MatrixType* matrix;
+#ifdef HAVE_CUDA
+ cusparseHandle_t* cusparseHandle;
+
+ cusparseMatDescr_t matrixDescriptor;
+#endif
+};
+
+
+template< typename Real >
+class CusparseCSR
+{};
+
+template<>
+class CusparseCSR< double > : public CusparseCSRBase< double >
+{
+ public:
+
+ template< typename InVector,
+ typename OutVector >
+ void vectorProduct( const InVector& inVector,
+ OutVector& outVector ) const
+ {
+ TNL_ASSERT_TRUE( matrix, "matrix was not initialized" );
+#ifdef HAVE_CUDA
+ double d = 1.0;
+ double* alpha = &d;
+ cusparseDcsrmv( *( this->cusparseHandle ),
+ CUSPARSE_OPERATION_NON_TRANSPOSE,
+ this->matrix->getRows(),
+ this->matrix->getColumns(),
+ this->matrix->getValues().getSize(),
+ alpha,
+ this->matrixDescriptor,
+ this->matrix->getValues().getData(),
+ this->matrix->getRowPointers().getData(),
+ this->matrix->getColumnIndexes().getData(),
+ inVector.getData(),
+ alpha,
+ outVector.getData() );
+#endif
+ }
+};
+
+template<>
+class CusparseCSR< float > : public CusparseCSRBase< float >
+{
+ public:
+
+ template< typename InVector,
+ typename OutVector >
+ void vectorProduct( const InVector& inVector,
+ OutVector& outVector ) const
+ {
+ TNL_ASSERT_TRUE( matrix, "matrix was not initialized" );
+#ifdef HAVE_CUDA
+ float d = 1.0;
+ float* alpha = &d;
+ cusparseScsrmv( *( this->cusparseHandle ),
+ CUSPARSE_OPERATION_NON_TRANSPOSE,
+ this->matrix->getRows(),
+ this->matrix->getColumns(),
+ this->matrix->getValues().getSize(),
+ alpha,
+ this->matrixDescriptor,
+ this->matrix->getValues().getData(),
+ this->matrix->getRowPointers().getData(),
+ this->matrix->getColumnIndexes().getData(),
+ inVector.getData(),
+ alpha,
+ outVector.getData() );
+#endif
+ }
+};
+
+} // namespace TNL
\ No newline at end of file
--
GitLab