hemv.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_HEMV_HH
#define TLAPACK_LEGACY_HEMV_HH
 
#include "tlapack/blas/hemv.hpp"
#include "tlapack/legacy_api/base/types.hpp"
#include "tlapack/legacy_api/base/utils.hpp"
 
namespace tlapack {
namespace legacy {
 
    template <typename TA, typename TX, typename TY>
    void hemv(Layout layout,
              Uplo uplo,
              idx_t n,
              scalar_type<TA, TX, TY> alpha,
              TA const* A,
              idx_t lda,
              TX const* x,
              int_t incx,
              scalar_type<TA, TX, TY> beta,
              TY* y,
              int_t incy)
    {
        using internal::create_matrix;
        using internal::create_rowmajor_matrix;
        using scalar_t = scalar_type<TA, TX, TY>;
 
        // check arguments
        tlapack_check_false(layout != Layout::ColMajor &&
                            layout != Layout::RowMajor);
        tlapack_check_false(uplo != Uplo::Lower && uplo != Uplo::Upper);
        tlapack_check_false(n < 0);
        tlapack_check_false(lda < n);
        tlapack_check_false(incx == 0);
        tlapack_check_false(incy == 0);
 
        // quick return
        if (n == 0 || ((alpha == scalar_t(0)) && (beta == scalar_t(1)))) return;
 
        if (alpha == scalar_t(0)) {
            tlapack_expr_with_vector(
                y_, TY, n, y, incy,
                if (beta == scalar_t(0)) for (idx_t i = 0; i < n; ++i) y_[i] =
                    TY(0);
                else for (idx_t i = 0; i < n; ++i) y_[i] *= beta);
        }
        else {
            tlapack_expr_with_2vectors(
                x_, TX, n, x, incx, y_, TY, n, y, incy,
                if (beta ==
                    scalar_t(
                        0)) if (layout ==
                                Layout::ColMajor) return hemv(uplo, alpha,
                                                              create_matrix<TA>(
                                                                  (TA*)A, n, n,
                                                                  lda),
                                                              x_, y_);
                else return hemv(uplo, alpha,
                                 create_rowmajor_matrix<TA>((TA*)A, n, n, lda),
                                 x_, y_);
                else if (layout == Layout::ColMajor) return hemv(
                    uplo, alpha, create_matrix<TA>((TA*)A, n, n, lda), x_, beta,
                    y_);
                else return hemv(uplo, alpha,
                                 create_rowmajor_matrix<TA>((TA*)A, n, n, lda),
                                 x_, beta, y_));
        }
    }
 
}  // namespace legacy
}  // namespace tlapack
 
#endif  //  #ifndef TLAPACK_LEGACY_HEMV_HH