cu_linear_operator.h

Go to the documentation of this file.

/*
 *  SPDX-FileCopyrightText: Copyright 2021, Siavash Ameli <sameli@berkeley.edu>
 *  SPDX-License-Identifier: BSD-3-Clause
 *  SPDX-FileType: SOURCE
 *
 *  This program is free software: you can redistribute it and/or modify it
 *  under the terms of the license found in the LICENSE.txt file in the root
 *  directory of this source tree.
 */
 
 
#ifndef _CU_LINEAR_OPERATOR_CU_LINEAR_OPERATOR_H_
#define _CU_LINEAR_OPERATOR_CU_LINEAR_OPERATOR_H_
 
// =======
// Headers
// =======
 
#include <cublas_v2.h>  // cublasHandle_t
#include <cusparse.h>  // cusparseHandle_t
#include "../_definitions/types.h"  // FlagType, IndexType, LongIndexType
#include "../_c_linear_operator/c_linear_operator.h"  // cLinearOperator
 
 
// ==================
// cu Linear Operator
// ==================
 
 
template <typename DataType>
class cuLinearOperator: virtual public cLinearOperator<DataType>
{
    public:
 
        // Member methods
        cuLinearOperator();
        explicit cuLinearOperator(int num_gpu_devices_);
 
        virtual ~cuLinearOperator();
 
        cublasHandle_t get_cublas_handle() const;
 
    protected:
 
        // Member methods
        int query_gpu_devices() const;
        void initialize_cublas_handle();
        void initialize_cusparse_handle();
 
        // Member data
        int num_gpu_devices;
        bool copied_host_to_device;
        cublasHandle_t* cublas_handle;
        cusparseHandle_t* cusparse_handle;
};
 
#endif  // _CU_LINEAR_OPERATOR_CU_LINEAR_OPERATOR_H_