cublasHandle.hpp

Go to the documentation of this file.

/** \file cublasHandle.hpp
 * \author Jared R. Males
 * \brief Management of a cublas handle
 * \ingroup cuda_files
 *
 */
 
//***********************************************************************//
// Copyright 2025 Jared R. Males (jaredmales@gmail.com)
//
// This file is part of mxlib.
//
// mxlib is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// mxlib is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with mxlib.  If not, see <http://www.gnu.org/licenses/>.
//***********************************************************************//
 
#ifndef math_cublasHandle_hpp
#define math_cublasHandle_hpp
 
#ifdef MXLIB_CUDA
 
#include <cuda_runtime.h>
#include <cublas_v2.h>
 
namespace mx
{
namespace cuda
{
 
/// Management of a cublas handle
/** RAII management of a cublas handle.
 *
 * The handle is not created automatically on default construction, e.g. in case
 * it is desired to do so in a critical block scope.
 *
 * The handle is destroyed on the call to this class's destructor.
 *
 * \todo throw exceptions in cuda::cublasHandle
 */
struct cublasHandle
{
 
  private:
    cublasHandle_t m_handle{ NULL };
 
  public:
    /// Default c'tor
    /** Does not create the handle.
     */
    cublasHandle()
    {
    }
 
    /// Constructor with option to create / not create the handle
    explicit cublasHandle( bool create /**< [in] if true the handle is created. if false it is not created. */ )
    {
        if( create )
        {
            cublasStatus_t cbec = this->create();
 
            if( cbec != CUBLAS_STATUS_SUCCESS )
            {
                std::string msg = std::format( "cublasHandle::cublasHandle error from create: [{}] {}\n",
                                          cublasGetStatusName( cbec ),
                                          cublasGetStatusString( cbec ) );
 
                throw mx::exception<verbose::d>( mx::error_t::liberr, msg);
            }
        }
    }
 
    /// Destructor
    ~cublasHandle()
    {
        cublasStatus_t cbec = destroy();
        if( cbec != CUBLAS_STATUS_SUCCESS )
        {
            std::cerr << std::format( "cublasHandle::~cublasHandle error from destroy: [{}] {}\n",
                                      cublasGetStatusName( cbec ),
                                      cublasGetStatusString( cbec ) );
        }
    }
 
    /// Create (allocate) the handle.
    /**
     *  \returns the cuBLAS status code from cublasDestroy or cublasCreate
     */
    cublasStatus_t create()
    {
 
        cublasStatus_t cbec = destroy();
        if( cbec != CUBLAS_STATUS_SUCCESS )
        {
            return cbec;
        }
 
        cbec = cublasCreate( &m_handle );
 
        if( cbec != CUBLAS_STATUS_SUCCESS )
        {
            destroy(); // we try but ignore any errors.  Hoping that any possible cleanup occurs, and sets nullptr if
                       // needed.
        }
 
        return cbec;
    }
 
    /// Destroy (de-allocate) the handle.
    /**
     *  \returns the cuBLAS status code from cublasDestroy
     */
    cublasStatus_t destroy()
    {
        cublasStatus_t cbec = CUBLAS_STATUS_SUCCESS;
        if( m_handle )
        {
            cublasStatus_t cbec = cublasDestroy( m_handle );
 
            m_handle = nullptr;
        }
 
        return cbec;
    }
 
    /// Get the handle for use in calls to cublas routines
    /**
     * \returns the cublas handle
     */
    cublasHandle_t operator()()
    {
        return m_handle;
    }
 
    /// Conversion operator, allows objects of this class to be used as if they are the handle
    /**
     * \returns the cublas handle
     */
    operator cublasHandle_t()
    {
        return m_handle;
    }
};
 
} // namespace cuda
} // namespace mx
 
#endif // MXLIB_CUDA
 
#endif // math_cublasHandle_hpp