laset.hpp

Go to the documentation of this file.

 
//
// 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_LASET_HH
#define TLAPACK_LEGACY_LASET_HH
 
#include "tlapack/lapack/laset.hpp"
 
namespace tlapack {
namespace legacy {
 
    template <class uplo_t, typename TA>
    void laset(
        uplo_t uplo, idx_t m, idx_t n, TA alpha, TA beta, TA* A, idx_t lda)
    {
        using internal::create_matrix;
 
        // check arguments
        tlapack_check_false(uplo != Uplo::Lower && uplo != Uplo::Upper &&
                            uplo != Uplo::General);
 
        // quick return
        if (m <= 0 || n <= 0) return;
 
        // Matrix views
        auto A_ = create_matrix<TA>(A, m, n, lda);
 
        return laset(uplo, alpha, beta, A_);
    }
 
    template <typename TA>
    void laset(MatrixType matrixtype,
               idx_t m,
               idx_t n,
               TA alpha,
               TA beta,
               TA* A,
               idx_t lda)
    {
        if (matrixtype == MatrixType::Upper) {
            laset(UPPER_TRIANGLE, m, n, alpha, beta, A, lda);
        }
        else if (matrixtype == MatrixType::Lower) {
            laset(LOWER_TRIANGLE, m, n, alpha, beta, A, lda);
        }
        else {
            laset(GENERAL, m, n, alpha, beta, A, lda);
        }
    }
 
}  // namespace legacy
}  // namespace tlapack
 
#endif  // TLAPACK_LEGACY_LASET_HH