cublas_types.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 _CUDA_DYNAMIC_LOADING_CUBLAS_TYPES_H_
#define _CUDA_DYNAMIC_LOADING_CUBLAS_TYPES_H_
 
// =======
// Headers
// =======
 
// Avoid CUBLAS numeration value not handled in switch [-Wswitch-enum] warning
#ifdef _MSC_VER
    #pragma warning(push, 0)  // Suppress all warnings from the followings
    #include <cublas_v2.h>
    #pragma warning(pop)  // Restore previous warning level
#elif defined(__INTEL_LLVM_COMPILER) || defined(__INTEL_COMPILER)
    #pragma warning(push, 0)
    #include <cublas_v2.h>
    #pragma warning(pop)
#elif defined(__GNUC__) || defined(__clang__)
    #pragma GCC diagnostic push
    #pragma GCC diagnostic ignored "-Wswitch-enum"
    #include <cublas_v2.h>
    #pragma GCC diagnostic pop
#else
    #include <cublas_v2.h>  // cublasSgemv, cublasDgemv, cublasScopy,
                            // cublasDcopy, cublasSaxpy, cublasDaxpy,
                            // cublasSdot, cublasDdot, cublasSnrm2,
                            // cublasDnrm2, cublasSscal, cublasDscal
                            // cublasHandle_t, cublasStatus_t,
                            // cublasSetMathMode
#endif
 
// =====
// Types
// =====
 
// cublasCreate
typedef cublasStatus_t (*cublasCreate_type)(cublasHandle_t* handle);
 
// cublasDestroy
typedef cublasStatus_t (*cublasDestroy_type)(cublasHandle_t handle);
 
// cublasSetMathMode
typedef cublasStatus_t (*cublasSetMathMode_type)(
        cublasHandle_t handle,
        cublasMath_t mode);
 
// cublasSgemv
typedef cublasStatus_t (*cublasSgemv_type)(
        cublasHandle_t handle,
        cublasOperation_t trans,
        int m,
        int n,
        const float* alpha,
        const float* A,
        int lda,
        const float* x,
        int incx,
        const float* beta,
        float* y,
        int incy);
 
// cublasDgemv
typedef cublasStatus_t (*cublasDgemv_type)(
        cublasHandle_t handle,
        cublasOperation_t trans,
        int m,
        int n,
        const double* alpha,
        const double* A,
        int lda,
        const double* x,
        int incx,
        const double* beta,
        double* y,
        int incy);
 
// cublasScopy
typedef cublasStatus_t (*cublasScopy_type)(
        cublasHandle_t handle, int n,
        const float* x,
        int incx,
        float* y,
        int incy);
 
// cublasDcopy
typedef cublasStatus_t (*cublasDcopy_type)(
        cublasHandle_t handle, int n,
        const double* x,
        int incx,
        double* y,
        int incy);
 
// cublasSaxpy
typedef cublasStatus_t (*cublasSaxpy_type)(
        cublasHandle_t handle,
        int n,
        const float *alpha,
        const float *x,
        int incx,
        float *y,
        int incy);
 
// cublasDaxpy
typedef cublasStatus_t (*cublasDaxpy_type)(
        cublasHandle_t handle,
        int n,
        const double *alpha,
        const double *x,
        int incx,
        double *y,
        int incy);
 
// cublasSdot
typedef cublasStatus_t (*cublasSdot_type)(
        cublasHandle_t handle,
        int n,
        const float *x,
        int incx,
        const float *y,
        int incy,
        float *result);
 
// cublasDdot
typedef cublasStatus_t (*cublasDdot_type)(
        cublasHandle_t handle,
        int n,
        const double *x,
        int incx,
        const double *y,
        int incy,
        double *result);
 
// cublasSnrm2
typedef cublasStatus_t (*cublasSnrm2_type)(
        cublasHandle_t handle,
        int n,
        const float *x,
        int incx,
        float *result);
 
// cublasDnrm2
typedef cublasStatus_t (*cublasDnrm2_type)(
        cublasHandle_t handle,
        int n,
        const double *x,
        int incx,
        double *result);
 
// cublasSscal
typedef cublasStatus_t (*cublasSscal_type)(
        cublasHandle_t handle,
        int n,
        const float *alpha,
        float *x,
        int incx);
 
// cublasDscal
typedef cublasStatus_t (*cublasDscal_type)(
        cublasHandle_t handle,
        int n,
        const double *alpha,
        double *x,
        int incx);
 
#endif  // _CUDA_DYNAMIC_LOADING_CUBLAS_TYPES_H_