apxdby.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_EGBLAS_MODE

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

#include <egblas.hpp>

#endif

namespace etl::impl::egblas {

#ifdef EGBLAS_HAS_SAPXDBY
static constexpr bool has_sapxdby = true;
#else
static constexpr bool has_sapxdby = false;
#endif

inline void apxdby([[maybe_unused]] size_t n,
                   [[maybe_unused]] float alpha,
                   [[maybe_unused]] float* A,
                   [[maybe_unused]] size_t lda,
                   [[maybe_unused]] float beta,
                   [[maybe_unused]] float* B,
                   [[maybe_unused]] size_t ldb) {
#ifdef EGBLAS_HAS_SAPXDBY
    inc_counter("egblas");
    egblas_sapxdby(n, alpha, A, lda, beta, B, ldb);
#else
    cpp_unreachable("Invalid call to egblas::apxdby");
#endif
}

#ifdef EGBLAS_HAS_DAPXDBY
static constexpr bool has_dapxdby = true;
#else
static constexpr bool has_dapxdby = false;
#endif

inline void apxdby([[maybe_unused]] size_t n,
                   [[maybe_unused]] double alpha,
                   [[maybe_unused]] double* A,
                   [[maybe_unused]] size_t lda,
                   [[maybe_unused]] double beta,
                   [[maybe_unused]] double* B,
                   [[maybe_unused]] size_t ldb) {
#ifdef EGBLAS_HAS_DAPXDBY
    inc_counter("egblas");
    egblas_dapxdby(n, alpha, A, lda, beta, B, ldb);
#else
    cpp_unreachable("Invalid call to egblas::apxdby");
#endif
}

#ifdef EGBLAS_HAS_CAPXDBY
static constexpr bool has_capxdby = true;
#else
static constexpr bool has_capxdby = false;
#endif

inline void apxdby([[maybe_unused]] size_t n,
                   [[maybe_unused]] std::complex<float> alpha,
                   [[maybe_unused]] std::complex<float>* A,
                   [[maybe_unused]] size_t lda,
                   [[maybe_unused]] std::complex<float> beta,
                   [[maybe_unused]] std::complex<float>* B,
                   [[maybe_unused]] size_t ldb) {
#ifdef EGBLAS_HAS_CAPXDBY
    inc_counter("egblas");
    egblas_capxdby(n, complex_cast(alpha), reinterpret_cast<cuComplex*>(A), lda, complex_cast(beta), reinterpret_cast<cuComplex*>(B), ldb);
#else
    cpp_unreachable("Invalid call to egblas::apxdby");
#endif
}

inline void apxdby([[maybe_unused]] size_t n,
                   [[maybe_unused]] etl::complex<float> alpha,
                   [[maybe_unused]] etl::complex<float>* A,
                   [[maybe_unused]] size_t lda,
                   [[maybe_unused]] std::complex<float> beta,
                   [[maybe_unused]] etl::complex<float>* B,
                   [[maybe_unused]] size_t ldb) {
#ifdef EGBLAS_HAS_CAPXDBY
    inc_counter("egblas");
    egblas_capxdby(n, complex_cast(alpha), reinterpret_cast<cuComplex*>(A), lda, complex_cast(beta), reinterpret_cast<cuComplex*>(B), ldb);
#else
    cpp_unreachable("Invalid call to egblas::apxdby");
#endif
}

#ifdef EGBLAS_HAS_ZAPXDBY
static constexpr bool has_zapxdby = true;
#else
static constexpr bool has_zapxdby = false;
#endif

inline void apxdby([[maybe_unused]] size_t n,
                   [[maybe_unused]] std::complex<double> alpha,
                   [[maybe_unused]] std::complex<double>* A,
                   [[maybe_unused]] size_t lda,
                   [[maybe_unused]] std::complex<double> beta,
                   [[maybe_unused]] std::complex<double>* B,
                   [[maybe_unused]] size_t ldb) {
#ifdef EGBLAS_HAS_ZAPXDBY
    inc_counter("egblas");
    egblas_zapxdby(n, complex_cast(alpha), reinterpret_cast<cuDoubleComplex*>(A), lda, complex_cast(beta), reinterpret_cast<cuDoubleComplex*>(B), ldb);
#else
    cpp_unreachable("Invalid call to egblas::apxdby");
#endif
}

inline void apxdby([[maybe_unused]] size_t n,
                   [[maybe_unused]] etl::complex<double> alpha,
                   [[maybe_unused]] etl::complex<double>* A,
                   [[maybe_unused]] size_t lda,
                   [[maybe_unused]] std::complex<double> beta,
                   [[maybe_unused]] etl::complex<double>* B,
                   [[maybe_unused]] size_t ldb) {
#ifdef EGBLAS_HAS_ZAPXDBY
    inc_counter("egblas");
    egblas_zapxdby(n, complex_cast(alpha), reinterpret_cast<cuDoubleComplex*>(A), lda, complex_cast(beta), reinterpret_cast<cuDoubleComplex*>(B), ldb);
#else
    cpp_unreachable("Invalid call to egblas::apxdby");
#endif
}

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