libdpf/include/grotto/gadgets/activations/hardswish.hpp

/// @file grotto/gadgets/activations/hardswish.hpp
/// @author Ryan Henry <ryan.henry@ucalgary.ca>
/// @brief Cleartext reference for `hardswish`.
/// @copyright Copyright (c) 2019-2023 Ryan Henry and others
/// @license Released under a GNU General Public v2.0 (GPLv2) license;
///          see [LICENSE.md](@ref license) for details.

#ifndef LIBDPF_INCLUDE_GROTTO_GADGETS_ACTIVATIONS_HARDSWISH_HPP__
#define LIBDPF_INCLUDE_GROTTO_GADGETS_ACTIVATIONS_HARDSWISH_HPP__

#include <cmath>
#include <array>
#include <limits>

#include "grotto/gadget_hints.hpp"

namespace grotto
{

namespace gadgets
{

HEDLEY_DIAGNOSTIC_PUSH
HEDLEY_DIAGNOSTIC_DISABLE_DEPRECATED


/// @brief Cleartext `hardswish`.
/// @see grotto::make_hardswish_lut
/// @note Straight-line cleartext. Fixed-point evaluation is the symbol in `@see` when that symbol is a LUT or an `eval_*` function.
/// \complexity One evaluation of `operator()` as written in this header. There is no domain scan and no loop. `Θ(1)` time and extra space relative to that expression.
struct HEDLEY_DEPRECATED_FOR(2026, grotto::make_hardswish_lut) hardswish
{
    /// @brief Evaluate the cleartext map.
    /// @tparam T argument and result type
    /// @param x the input
    /// @return the cleartext image of `x`
    /// \complexity The body is straight-line. `Θ(1)` time and extra space.
    template <typename T>
    T operator()(T x)
    {
        if (x <= -3) return 0;
        if (x >= 3) return x;
        return x*(x+3)/6;
    }
};

template <>
struct gadget_hints<hardswish>
{
    static constexpr double min = std::numeric_limits<double>::min();
    static constexpr double max = std::numeric_limits<double>::max();
    static constexpr double poles[] = { };
    static constexpr double interesting_points[] = { -3, 3 };
    static constexpr unsigned degree = 2;
    static constexpr bool has_canonical_representation = true;
    static constexpr double canonical_bounds[] = { -3, 3 };
    static constexpr std::array<double, degree+1> canonical_polys[] = { {0}, {0,0.5,1/6.0}, {0,1,0} };
};

HEDLEY_DIAGNOSTIC_POP

}  // namespace gadgets

}  // namespace grotto

#endif  // LIBDPF_INCLUDE_GROTTO_GADGETS_ACTIVATIONS_HARDSWISH_HPP__