clip_value.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 "etl/util/complex_cast.hpp"
#include "etl/util/safe_cast.hpp"

#include <egblas.hpp>

#endif

namespace etl::impl::egblas {

#ifdef EGBLAS_HAS_SCLIP_VALUE
static constexpr bool has_sclip_value = true;
#else
static constexpr bool has_sclip_value = false;
#endif

inline void clip_value([[maybe_unused]] size_t n,
                       [[maybe_unused]] const float alpha,
                       [[maybe_unused]] float A,
                       [[maybe_unused]] float B,
                       [[maybe_unused]] float* C,
                       [[maybe_unused]] size_t ldc) {
#ifdef EGBLAS_HAS_SCLIP_VALUE
    inc_counter("egblas");
    egblas_sclip_value(n, alpha, A, B, C, ldc);
#else
    cpp_unreachable("Invalid call to egblas::clip_value");
#endif
}

#ifdef EGBLAS_HAS_DCLIP_VALUE
static constexpr bool has_dclip_value = true;
#else
static constexpr bool has_dclip_value = false;
#endif

inline void clip_value([[maybe_unused]] size_t n,
                       [[maybe_unused]] double alpha,
                       [[maybe_unused]] double A,
                       [[maybe_unused]] double B,
                       [[maybe_unused]] double* C,
                       [[maybe_unused]] size_t ldc) {
#ifdef EGBLAS_HAS_DCLIP_VALUE
    inc_counter("egblas");
    egblas_dclip_value(n, alpha, A, B, C, ldc);
#else
    cpp_unreachable("Invalid call to egblas::clip_value");
#endif
}

#ifdef EGBLAS_HAS_CCLIP_VALUE
static constexpr bool has_cclip_value = true;
#else
static constexpr bool has_cclip_value = false;
#endif

inline void clip_value([[maybe_unused]] size_t n,
                       [[maybe_unused]] const std::complex<float> alpha,
                       [[maybe_unused]] std::complex<float> A,
                       [[maybe_unused]] std::complex<float> B,
                       [[maybe_unused]] std::complex<float>* C,
                       [[maybe_unused]] size_t ldc) {
#ifdef EGBLAS_HAS_CCLIP_VALUE
    inc_counter("egblas");
    egblas_cclip_value(n, complex_cast(alpha), complex_cast(A), complex_cast(B), reinterpret_cast<cuComplex*>(C), ldc);
#else
    cpp_unreachable("Invalid call to egblas::clip_value");
#endif
}

inline void clip_value([[maybe_unused]] size_t n,
                       [[maybe_unused]] const etl::complex<float> alpha,
                       [[maybe_unused]] etl::complex<float> A,
                       [[maybe_unused]] etl::complex<float> B,
                       [[maybe_unused]] etl::complex<float>* C,
                       [[maybe_unused]] size_t ldc) {
#ifdef EGBLAS_HAS_CCLIP_VALUE
    inc_counter("egblas");
    egblas_cclip_value(n, complex_cast(alpha), complex_cast(A), complex_cast(B), reinterpret_cast<cuComplex*>(C), ldc);
#else
    cpp_unreachable("Invalid call to egblas::clip_value");
#endif
}

#ifdef EGBLAS_HAS_ZCLIP_VALUE
static constexpr bool has_zclip_value = true;
#else
static constexpr bool has_zclip_value = false;
#endif

inline void clip_value([[maybe_unused]] size_t n,
                       [[maybe_unused]] const std::complex<double> alpha,
                       [[maybe_unused]] std::complex<double> A,
                       [[maybe_unused]] std::complex<double> B,
                       [[maybe_unused]] std::complex<double>* C,
                       [[maybe_unused]] size_t ldc) {
#ifdef EGBLAS_HAS_ZCLIP_VALUE
    inc_counter("egblas");
    egblas_zclip_value(n, complex_cast(alpha), complex_cast(A), complex_cast(B), reinterpret_cast<cuDoubleComplex*>(C), ldc);
#else
    cpp_unreachable("Invalid call to egblas::clip_value");
#endif
}

inline void clip_value([[maybe_unused]] size_t n,
                       [[maybe_unused]] const etl::complex<double> alpha,
                       [[maybe_unused]] etl::complex<double> A,
                       [[maybe_unused]] etl::complex<double> B,
                       [[maybe_unused]] etl::complex<double>* C,
                       [[maybe_unused]] size_t ldc) {
#ifdef EGBLAS_HAS_ZCLIP_VALUE
    inc_counter("egblas");
    egblas_zclip_value(n, complex_cast(alpha), complex_cast(A), complex_cast(B), reinterpret_cast<cuDoubleComplex*>(C), ldc);
#else
    cpp_unreachable("Invalid call to egblas::clip_value");
#endif
}

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