scalar_add.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_SCALAR_SADD

static constexpr bool has_scalar_sadd = true;

inline void scalar_add(float* x, size_t n, size_t s, const float beta) {
    inc_counter("egblas");
    egblas_scalar_sadd(x, n, s, beta);
}

#else

static constexpr bool has_scalar_sadd = false;

#endif

#ifdef EGBLAS_HAS_SCALAR_DADD

static constexpr bool has_scalar_dadd = true;

inline void scalar_add(double* x, size_t n, size_t s, const double beta) {
    inc_counter("egblas");
    egblas_scalar_dadd(x, n, s, beta);
}

#else

static constexpr bool has_scalar_dadd = false;

#endif

#ifdef EGBLAS_HAS_SCALAR_CADD

static constexpr bool has_scalar_cadd = true;

inline void scalar_add(etl::complex<float>* x, size_t n, size_t s, const etl::complex<float> beta) {
    inc_counter("egblas");
    egblas_scalar_cadd(reinterpret_cast<cuComplex*>(x), n, s, complex_cast(beta));
}

inline void scalar_add(std::complex<float>* x, size_t n, size_t s, const std::complex<float> beta) {
    inc_counter("egblas");
    egblas_scalar_cadd(reinterpret_cast<cuComplex*>(x), n, s, complex_cast(beta));
}

#else

static constexpr bool has_scalar_cadd = false;

#endif

#ifdef EGBLAS_HAS_SCALAR_ZADD

static constexpr bool has_scalar_zadd = true;

inline void scalar_add(etl::complex<double>* x, size_t n, size_t s, const etl::complex<double> beta) {
    inc_counter("egblas");
    egblas_scalar_zadd(reinterpret_cast<cuDoubleComplex*>(x), n, s, complex_cast(beta));
}

inline void scalar_add(std::complex<double>* x, size_t n, size_t s, const std::complex<double> beta) {
    inc_counter("egblas");
    egblas_scalar_zadd(reinterpret_cast<cuDoubleComplex*>(x), n, s, complex_cast(beta));
}

#else

static constexpr bool has_scalar_zadd = false;

#endif

#ifndef ETL_EGBLAS_MODE

template <typename T>
inline void scalar_add([[maybe_unused]] T* x, [[maybe_unused]] size_t n, [[maybe_unused]] size_t s, [[maybe_unused]] const T beta) {
    cpp_unreachable("Invalid call to egblas::scalar_add");
}

#endif

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