blas1_add.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSL-1.0
// SPDX-FileCopyrightText: Copyright (C) 2023 Tobias Hienzsch
 
#ifndef TETL_LINALG_BLAS1_ADD_HPP
#define TETL_LINALG_BLAS1_ADD_HPP
 
#include <etl/_contracts/check.hpp>
#include <etl/_linalg/exposition.hpp>
 
namespace etl::linalg {
 
/// \ingroup linalg
template <in_object InObj1, in_object InObj2, out_object OutObj>
    requires(InObj1::rank() == OutObj::rank() and InObj2::rank() == OutObj::rank())
constexpr auto add(InObj1 x, InObj2 y, OutObj z) -> void
{
    static_assert(detail::possibly_addable<InObj1, InObj2, OutObj>());
    TETL_PRECONDITION(detail::addable(x, y, z));
 
    using index_type = detail::common_index_type_t<InObj1, InObj2, OutObj>;
 
    if constexpr (OutObj::rank() == 1) {
        for (index_type row{0}; row < x.extent(0); ++row) {
            z(row) = x(row) + y(row);
        }
    } else {
        static_assert(OutObj::rank() == 2);
        for (index_type row{0}; row < x.extent(0); ++row) {
            for (index_type col{0}; col < x.extent(1); ++col) {
                z(row, col) = x(row, col) + y(row, col);
            }
        }
    }
}
 
} // namespace etl::linalg
 
#endif // TETL_LINALG_BLAS1_ADD_HPP