axpy.hpp

Go to the documentation of this file.

//=======================================================================
// Copyright (c) 2014-2023 Baptiste Wicht
// Distributed under the terms of the MIT License.
// (See accompanying file LICENSE or copy at
//  http://opensource.org/licenses/MIT)
//=======================================================================

#pragma once

#ifdef ETL_CUBLAS_MODE

#include "etl/impl/cublas/cuda.hpp"
#include "etl/impl/cublas/cublas.hpp"

#endif

namespace etl::impl::cublas {

#ifdef ETL_CUBLAS_MODE

inline void cublas_axpy(cublasHandle_t handle, size_t n, const float* alpha, const float* A, size_t lda, float* B, size_t ldb) {
    cublas_check(cublasSaxpy(handle, n, alpha, A, lda, B, ldb));
}

inline void cublas_axpy(cublasHandle_t handle, size_t n, const double* alpha, const double* A, size_t lda, double* B, size_t ldb) {
    cublas_check(cublasDaxpy(handle, n, alpha, A, lda, B, ldb));
}

inline void cublas_axpy(
    cublasHandle_t handle, size_t n, const std::complex<float>* alpha, const std::complex<float>* A, size_t lda, std::complex<float>* B, size_t ldb) {
    cublas_check(
        cublasCaxpy(handle, n, reinterpret_cast<const cuComplex*>(alpha), reinterpret_cast<const cuComplex*>(A), lda, reinterpret_cast<cuComplex*>(B), ldb));
}

inline void cublas_axpy(
    cublasHandle_t handle, size_t n, const std::complex<double>* alpha, const std::complex<double>* A, size_t lda, std::complex<double>* B, size_t ldb) {
    cublas_check(cublasZaxpy(handle, n, reinterpret_cast<const cuDoubleComplex*>(alpha), reinterpret_cast<const cuDoubleComplex*>(A), lda,
                             reinterpret_cast<cuDoubleComplex*>(B), ldb));
}

inline void cublas_axpy(
    cublasHandle_t handle, size_t n, const etl::complex<float>* alpha, const etl::complex<float>* A, size_t lda, etl::complex<float>* B, size_t ldb) {
    cublas_check(
        cublasCaxpy(handle, n, reinterpret_cast<const cuComplex*>(alpha), reinterpret_cast<const cuComplex*>(A), lda, reinterpret_cast<cuComplex*>(B), ldb));
}

inline void cublas_axpy(
    cublasHandle_t handle, size_t n, const etl::complex<double>* alpha, const etl::complex<double>* A, size_t lda, etl::complex<double>* B, size_t ldb) {
    cublas_check(cublasZaxpy(handle, n, reinterpret_cast<const cuDoubleComplex*>(alpha), reinterpret_cast<const cuDoubleComplex*>(A), lda,
                             reinterpret_cast<cuDoubleComplex*>(B), ldb));
}

#endif

} //end of namespace etl::impl::cublas