axpy.hpp

Go to the documentation of this file.

 
//
// Copyright (c) 2017-2021, University of Tennessee. All rights reserved.
// Copyright (c) 2021-2023, University of Colorado Denver. All rights reserved.
//
// This file is part of <T>LAPACK.
// <T>LAPACK is free software: you can redistribute it and/or modify it under
// the terms of the BSD 3-Clause license. See the accompanying LICENSE file.
 
#ifndef TLAPACK_LEGACY_AXPY_HH
#define TLAPACK_LEGACY_AXPY_HH
 
#include "tlapack/blas/axpy.hpp"
#include "tlapack/legacy_api/base/types.hpp"
#include "tlapack/legacy_api/base/utils.hpp"
 
namespace tlapack {
namespace legacy {
 
    template <typename TX, typename TY>
    void axpy(idx_t n,
              scalar_type<TX, TY> alpha,
              TX const* x,
              int_t incx,
              TY* y,
              int_t incy)
    {
        tlapack_check_false(incx == 0);
        tlapack_check_false(incy == 0);
        using scalar_t = scalar_type<TX, TY>;
 
        // quick return
        if (n <= 0 || alpha == scalar_t(0)) return;
 
        tlapack_expr_with_2vectors(x_, TX, n, x, incx, y_, TY, n, y, incy,
                                   return axpy(alpha, x_, y_));
    }
 
}  // namespace legacy
}  // namespace tlapack
 
#endif  //  #ifndef TLAPACK_LEGACY_AXPY_HH