derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
909 lines
27 KiB
C++
909 lines
27 KiB
C++
/* **************************************************************************
|
|
* Copyright (C) 2020-2026 Advanced Micro Devices, Inc. All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
*
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
* SUCH DAMAGE.
|
|
* *************************************************************************/
|
|
|
|
#pragma once
|
|
|
|
#include "clientcommon.hpp"
|
|
#include "rocsolver_datatype2string.hpp"
|
|
#include <rocblas/rocblas.h>
|
|
|
|
/*!\file
|
|
* \brief provide template functions interfaces to BLAS and LAPACK interfaces,
|
|
* it is only used for testing not part of the GPU library
|
|
*/
|
|
|
|
/*template <typename T>
|
|
void cpu_iamax(rocblas_int n, const T *x, rocblas_int incx,
|
|
rocblas_int *result);
|
|
|
|
template <typename T>
|
|
void cpu_iamin(rocblas_int n, const T *x, rocblas_int incx,
|
|
rocblas_int *result);
|
|
|
|
template <typename T1, typename T2>
|
|
void cpu_asum(rocblas_int n, const T1 *x, rocblas_int incx, T2 *result);
|
|
|
|
template <typename T>
|
|
void cpu_copy(rocblas_int n, T *x, rocblas_int incx, T *y, rocblas_int incy);
|
|
|
|
template <typename T>
|
|
void cpu_dot(rocblas_int n, const T *x, rocblas_int incx, const T *y,
|
|
rocblas_int incy, T *result);
|
|
|
|
template <typename T1, typename T2>
|
|
void cpu_nrm2(rocblas_int n, const T1 *x, rocblas_int incx, T2 *result);
|
|
|
|
template <typename T>
|
|
void cpu_scal(rocblas_int n, const T alpha, T *x, rocblas_int incx);
|
|
|
|
template <typename T>
|
|
void cpu_swap(rocblas_int n, T *x, rocblas_int incx, T *y, rocblas_int incy);
|
|
|
|
template <typename T>
|
|
void cpu_ger(rocblas_int m, rocblas_int n, T alpha, T *x, rocblas_int incx,
|
|
T *y, rocblas_int incy, T *A, rocblas_int lda);
|
|
|
|
template <typename T>
|
|
void cpu_syr(rocblas_fill uplo, rocblas_int n, T alpha, T *x,
|
|
rocblas_int incx, T *A, rocblas_int lda);
|
|
*/
|
|
|
|
template <typename T, typename S>
|
|
S cpu_lange(char norm, rocblas_int m, rocblas_int n, T* A, rocblas_int lda, S* work);
|
|
|
|
template <typename T, typename S>
|
|
S cpu_gecon(char norm, rocblas_int n, T* A, rocblas_int lda, S anorm, T* work, S* rwork, rocblas_int* iwork);
|
|
|
|
template <typename T>
|
|
void cpu_axpy(rocblas_int n, T alpha, T* x, rocblas_int incx, T* y, rocblas_int incy);
|
|
|
|
template <typename T>
|
|
void cpu_gemv(rocblas_operation transA,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
T alpha,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* x,
|
|
rocblas_int incx,
|
|
T beta,
|
|
T* y,
|
|
rocblas_int incy);
|
|
|
|
template <typename T>
|
|
void cpu_gemm(rocblas_operation transA,
|
|
rocblas_operation transB,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T alpha,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb,
|
|
T beta,
|
|
T* C,
|
|
rocblas_int ldc);
|
|
|
|
template <typename T>
|
|
void cpu_symv_hemv(rocblas_fill uplo,
|
|
rocblas_int n,
|
|
T alpha,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* x,
|
|
rocblas_int incx,
|
|
T beta,
|
|
T* y,
|
|
rocblas_int incy);
|
|
|
|
template <typename T>
|
|
void cpu_symm_hemm(rocblas_side side,
|
|
rocblas_fill uplo,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
T alpha,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb,
|
|
T beta,
|
|
T* C,
|
|
rocblas_int ldc);
|
|
|
|
template <typename T>
|
|
void cpu_trsm(rocblas_side side,
|
|
rocblas_fill uplo,
|
|
rocblas_operation transA,
|
|
rocblas_diagonal diag,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
T alpha,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb);
|
|
/*
|
|
template <typename T>
|
|
void cpu_trsv(rocblas_fill uplo,
|
|
rocblas_operation transA,
|
|
rocblas_diagonal diag,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* x,
|
|
rocblas_int incx);
|
|
*/
|
|
template <typename T>
|
|
void cpu_trmm(rocblas_side side,
|
|
rocblas_fill uplo,
|
|
rocblas_operation transA,
|
|
rocblas_diagonal diag,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
T alpha,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb);
|
|
|
|
template <typename T>
|
|
void cpu_potf2(rocblas_fill uplo, rocblas_int n, T* A, rocblas_int lda, rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_potrf(rocblas_fill uplo, rocblas_int n, T* A, rocblas_int lda, rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_potrs(rocblas_fill uplo,
|
|
rocblas_int n,
|
|
rocblas_int nrhs,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb);
|
|
|
|
template <typename T>
|
|
void cpu_posv(rocblas_fill uplo,
|
|
rocblas_int n,
|
|
rocblas_int nrhs,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb,
|
|
rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_potri(rocblas_fill uplo, rocblas_int n, T* A, rocblas_int lda, rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_getf2(rocblas_int m, rocblas_int n, T* A, rocblas_int lda, rocblas_int* ipiv, rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_getrf(rocblas_int m, rocblas_int n, T* A, rocblas_int lda, rocblas_int* ipiv, rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_getrs(rocblas_operation trans,
|
|
rocblas_int n,
|
|
rocblas_int nrhs,
|
|
T* A,
|
|
rocblas_int lda,
|
|
rocblas_int* ipiv,
|
|
T* B,
|
|
rocblas_int ldb);
|
|
|
|
template <typename T>
|
|
void cpu_sytrs2(rocblas_fill uplo,
|
|
rocblas_int n,
|
|
rocblas_int nrhs,
|
|
T* A,
|
|
rocblas_int lda,
|
|
rocblas_int* ipiv,
|
|
T* B,
|
|
rocblas_int ldb);
|
|
|
|
template <typename T>
|
|
void cpu_sytrs(rocblas_fill uplo,
|
|
rocblas_int n,
|
|
rocblas_int nrhs,
|
|
T* A,
|
|
rocblas_int lda,
|
|
rocblas_int* ipiv,
|
|
T* B,
|
|
rocblas_int ldb);
|
|
|
|
template <typename T>
|
|
void cpu_getrs_npvt(rocblas_operation trans,
|
|
rocblas_int n,
|
|
rocblas_int nrhs,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb);
|
|
|
|
template <typename T>
|
|
void cpu_gesv(rocblas_int n,
|
|
rocblas_int nrhs,
|
|
T* A,
|
|
rocblas_int lda,
|
|
rocblas_int* ipiv,
|
|
T* B,
|
|
rocblas_int ldb,
|
|
rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_gels(rocblas_operation transR,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int nrhs,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb,
|
|
T* work,
|
|
rocblas_int lwork,
|
|
rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_getri(rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
rocblas_int* ipiv,
|
|
T* work,
|
|
rocblas_int lwork,
|
|
rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_trtri(rocblas_fill uplo,
|
|
rocblas_diagonal diag,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_larfg(rocblas_int n, T* alpha, T* x, rocblas_int incx, T* tau);
|
|
|
|
template <typename T>
|
|
void cpu_larf(rocblas_side side,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
T* x,
|
|
rocblas_int incx,
|
|
T* alpha,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* work);
|
|
|
|
template <typename T>
|
|
void cpu_larft(rocblas_direct direct,
|
|
rocblas_storev storev,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* V,
|
|
rocblas_int ldv,
|
|
T* tau,
|
|
T* F,
|
|
rocblas_int ldt);
|
|
|
|
template <typename T>
|
|
void cpu_larfb(rocblas_side side,
|
|
rocblas_operation trans,
|
|
rocblas_direct direct,
|
|
rocblas_storev storev,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* V,
|
|
rocblas_int ldv,
|
|
T* F,
|
|
rocblas_int ldt,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* W,
|
|
rocblas_int ldw);
|
|
|
|
template <typename T, typename SS>
|
|
void cpu_lasr(rocblas_side side,
|
|
rocblas_pivot pivot,
|
|
rocblas_direct direct,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
SS* C,
|
|
SS* S,
|
|
T* A,
|
|
rocblas_int lda);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_latrd(rocblas_fill uplo,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* A,
|
|
rocblas_int lda,
|
|
S* E,
|
|
T* tau,
|
|
T* W,
|
|
rocblas_int ldw);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_labrd(rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int nb,
|
|
T* A,
|
|
rocblas_int lda,
|
|
S* D,
|
|
S* E,
|
|
T* tauq,
|
|
T* taup,
|
|
T* X,
|
|
rocblas_int ldx,
|
|
T* Y,
|
|
rocblas_int ldy);
|
|
|
|
template <typename T, typename W>
|
|
void cpu_bdsqr(rocblas_fill uplo,
|
|
rocblas_int n,
|
|
rocblas_int nv,
|
|
rocblas_int nu,
|
|
rocblas_int nc,
|
|
W* D,
|
|
W* E,
|
|
T* V,
|
|
rocblas_int ldv,
|
|
T* U,
|
|
rocblas_int ldu,
|
|
T* C,
|
|
rocblas_int ldc,
|
|
W* work,
|
|
rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_geqr2(rocblas_int m, rocblas_int n, T* A, rocblas_int lda, T* ipiv, T* work);
|
|
|
|
template <typename T>
|
|
void cpu_geqrf(rocblas_int m, rocblas_int n, T* A, rocblas_int lda, T* ipiv, T* work, rocblas_int sizeW);
|
|
|
|
template <typename T>
|
|
void cpu_gerq2(rocblas_int m, rocblas_int n, T* A, rocblas_int lda, T* ipiv, T* work);
|
|
|
|
template <typename T>
|
|
void cpu_gerqf(rocblas_int m, rocblas_int n, T* A, rocblas_int lda, T* ipiv, T* work, rocblas_int sizeW);
|
|
|
|
template <typename T>
|
|
void cpu_geql2(rocblas_int m, rocblas_int n, T* A, rocblas_int lda, T* ipiv, T* work);
|
|
|
|
template <typename T>
|
|
void cpu_geqlf(rocblas_int m, rocblas_int n, T* A, rocblas_int lda, T* ipiv, T* work, rocblas_int sizeW);
|
|
|
|
template <typename T>
|
|
void cpu_gelq2(rocblas_int m, rocblas_int n, T* A, rocblas_int lda, T* ipiv, T* work);
|
|
|
|
template <typename T>
|
|
void cpu_gelqf(rocblas_int m, rocblas_int n, T* A, rocblas_int lda, T* ipiv, T* work, rocblas_int sizeW);
|
|
|
|
template <typename T>
|
|
void cpu_lacgv(rocblas_int n, T* x, rocblas_int incx);
|
|
|
|
template <typename T>
|
|
void cpu_lacpy(rocblas_fill uplo,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb);
|
|
|
|
template <typename T>
|
|
void cpu_laswp(rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
rocblas_int k1,
|
|
rocblas_int k2,
|
|
rocblas_int* ipiv,
|
|
rocblas_int inc);
|
|
|
|
template <typename T>
|
|
void cpu_org2r_ung2r(rocblas_int m, rocblas_int n, rocblas_int k, T* A, rocblas_int lda, T* Ipiv, T* work);
|
|
|
|
template <typename T>
|
|
void cpu_orgqr_ungqr(rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* Ipiv,
|
|
T* work,
|
|
rocblas_int sizeW);
|
|
|
|
template <typename T>
|
|
void cpu_orgl2_ungl2(rocblas_int m, rocblas_int n, rocblas_int k, T* A, rocblas_int lda, T* Ipiv, T* work);
|
|
|
|
template <typename T>
|
|
void cpu_orglq_unglq(rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* Ipiv,
|
|
T* work,
|
|
rocblas_int sizeW);
|
|
|
|
template <typename T>
|
|
void cpu_org2l_ung2l(rocblas_int m, rocblas_int n, rocblas_int k, T* A, rocblas_int lda, T* Ipiv, T* work);
|
|
|
|
template <typename T>
|
|
void cpu_orgql_ungql(rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* Ipiv,
|
|
T* work,
|
|
rocblas_int sizeW);
|
|
|
|
template <typename T>
|
|
void cpu_orgbr_ungbr(rocblas_storev storev,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* Ipiv,
|
|
T* work,
|
|
rocblas_int size_w);
|
|
|
|
template <typename T>
|
|
void cpu_orgtr_ungtr(rocblas_fill uplo,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* Ipiv,
|
|
T* work,
|
|
rocblas_int size_w);
|
|
|
|
template <typename T>
|
|
void cpu_orm2r_unm2r(rocblas_side side,
|
|
rocblas_operation trans,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* Ipiv,
|
|
T* C,
|
|
rocblas_int ldc,
|
|
T* work);
|
|
|
|
template <typename T>
|
|
void cpu_ormqr_unmqr(rocblas_side side,
|
|
rocblas_operation trans,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* Ipiv,
|
|
T* C,
|
|
rocblas_int ldc,
|
|
T* work,
|
|
rocblas_int sizeW);
|
|
|
|
template <typename T>
|
|
void cpu_orml2_unml2(rocblas_side side,
|
|
rocblas_operation trans,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* Ipiv,
|
|
T* C,
|
|
rocblas_int ldc,
|
|
T* work);
|
|
|
|
template <typename T>
|
|
void cpu_ormlq_unmlq(rocblas_side side,
|
|
rocblas_operation trans,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* Ipiv,
|
|
T* C,
|
|
rocblas_int ldc,
|
|
T* work,
|
|
rocblas_int sizeW);
|
|
|
|
template <typename T>
|
|
void cpu_lauum(rocblas_fill uploR, rocblas_int n, T* A, rocblas_int lda);
|
|
|
|
template <typename T>
|
|
void cpu_orm2l_unm2l(rocblas_side side,
|
|
rocblas_operation trans,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* Ipiv,
|
|
T* C,
|
|
rocblas_int ldc,
|
|
T* work);
|
|
|
|
template <typename T>
|
|
void cpu_ormql_unmql(rocblas_side side,
|
|
rocblas_operation trans,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* Ipiv,
|
|
T* C,
|
|
rocblas_int ldc,
|
|
T* work,
|
|
rocblas_int sizeW);
|
|
|
|
template <typename T>
|
|
void cpu_ormbr_unmbr(rocblas_storev storev,
|
|
rocblas_side side,
|
|
rocblas_operation trans,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
rocblas_int k,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* Ipiv,
|
|
T* C,
|
|
rocblas_int ldc,
|
|
T* work,
|
|
rocblas_int sizeW);
|
|
|
|
template <typename T>
|
|
void cpu_ormtr_unmtr(rocblas_side side,
|
|
rocblas_fill uplo,
|
|
rocblas_operation trans,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* Ipiv,
|
|
T* C,
|
|
rocblas_int ldc,
|
|
T* work,
|
|
rocblas_int sizeW);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_gebd2(rocblas_int m, rocblas_int n, T* A, rocblas_int lda, S* D, S* E, T* tauq, T* taup, T* work);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_gebrd(rocblas_int m,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
S* D,
|
|
S* E,
|
|
T* tauq,
|
|
T* taup,
|
|
T* work,
|
|
rocblas_int size_w);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_sytrd_hetrd(rocblas_fill uplo,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
S* D,
|
|
S* E,
|
|
T* tau,
|
|
T* work,
|
|
rocblas_int size_w);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_sytd2_hetd2(rocblas_fill uplo, rocblas_int n, T* A, rocblas_int lda, S* D, S* E, T* tau);
|
|
|
|
template <typename T, typename W>
|
|
void cpu_gesvd(rocblas_svect leftv,
|
|
rocblas_svect rightv,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
W* S,
|
|
T* U,
|
|
rocblas_int ldu,
|
|
T* V,
|
|
rocblas_int ldv,
|
|
T* work,
|
|
rocblas_int lwork,
|
|
W* rwork,
|
|
rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_sterf(rocblas_int n, T* D, T* E);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_steqr(rocblas_evect evect,
|
|
rocblas_int n,
|
|
S* D,
|
|
S* E,
|
|
T* C,
|
|
rocblas_int ldc,
|
|
S* work,
|
|
rocblas_int* info);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_stedc(rocblas_evect evect,
|
|
rocblas_int n,
|
|
S* D,
|
|
S* E,
|
|
T* C,
|
|
rocblas_int ldc,
|
|
T* work,
|
|
rocblas_int lwork,
|
|
S* rwork,
|
|
rocblas_int lrwork,
|
|
rocblas_int* iwork,
|
|
rocblas_int liwork,
|
|
rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_stebz(rocblas_erange erange,
|
|
rocblas_eorder eorder,
|
|
rocblas_int n,
|
|
T vl,
|
|
T vu,
|
|
rocblas_int il,
|
|
rocblas_int iu,
|
|
T abstol,
|
|
T* D,
|
|
T* E,
|
|
rocblas_int* nev,
|
|
rocblas_int* nsplit,
|
|
T* W,
|
|
rocblas_int* iblock,
|
|
rocblas_int* isplit,
|
|
T* work,
|
|
rocblas_int* iwork,
|
|
rocblas_int* info);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_stein(rocblas_int n,
|
|
S* D,
|
|
S* E,
|
|
rocblas_int* nev,
|
|
S* W,
|
|
rocblas_int* iblock,
|
|
rocblas_int* isplit,
|
|
T* Z,
|
|
rocblas_int ldz,
|
|
S* work,
|
|
rocblas_int* iwork,
|
|
rocblas_int* ifail,
|
|
rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_sygs2_hegs2(rocblas_eform itype,
|
|
rocblas_fill uplo,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb);
|
|
|
|
template <typename T>
|
|
void cpu_sygst_hegst(rocblas_eform itype,
|
|
rocblas_fill uplo,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_syev_heev(rocblas_evect evect,
|
|
rocblas_fill uplo,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
S* W,
|
|
T* work,
|
|
rocblas_int lwork,
|
|
S* rwork,
|
|
rocblas_int lrwork,
|
|
rocblas_int* info);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_syevd_heevd(rocblas_evect evect,
|
|
rocblas_fill uplo,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
S* W,
|
|
T* work,
|
|
rocblas_int lwork,
|
|
S* rwork,
|
|
rocblas_int lrwork,
|
|
rocblas_int* iwork,
|
|
rocblas_int liwork,
|
|
rocblas_int* info);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_syevx_heevx(rocblas_evect evect,
|
|
rocblas_erange erange,
|
|
rocblas_fill uplo,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
S vl,
|
|
S vu,
|
|
rocblas_int il,
|
|
rocblas_int iu,
|
|
S abstol,
|
|
rocblas_int* nev,
|
|
S* W,
|
|
T* Z,
|
|
rocblas_int ldz,
|
|
T* work,
|
|
rocblas_int lwork,
|
|
S* rwork,
|
|
rocblas_int* iwork,
|
|
rocblas_int* ifail,
|
|
rocblas_int* info);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_sygv_hegv(rocblas_eform itype,
|
|
rocblas_evect evect,
|
|
rocblas_fill uplo,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb,
|
|
S* W,
|
|
T* work,
|
|
rocblas_int lwork,
|
|
S* rwork,
|
|
rocblas_int* info);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_sygvd_hegvd(rocblas_eform itype,
|
|
rocblas_evect evect,
|
|
rocblas_fill uplo,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb,
|
|
S* W,
|
|
T* work,
|
|
rocblas_int lwork,
|
|
S* rwork,
|
|
rocblas_int lrwork,
|
|
rocblas_int* iwork,
|
|
rocblas_int liwork,
|
|
rocblas_int* info);
|
|
|
|
template <typename T, typename S>
|
|
void cpu_sygvx_hegvx(rocblas_eform itype,
|
|
rocblas_evect evect,
|
|
rocblas_erange erange,
|
|
rocblas_fill uplo,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
T* B,
|
|
rocblas_int ldb,
|
|
S vl,
|
|
S vu,
|
|
rocblas_int il,
|
|
rocblas_int iu,
|
|
S abstol,
|
|
rocblas_int* m,
|
|
S* W,
|
|
T* Z,
|
|
rocblas_int ldz,
|
|
T* work,
|
|
rocblas_int lwork,
|
|
S* rwork,
|
|
rocblas_int* iwork,
|
|
rocblas_int* ifail,
|
|
rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_lasyf(rocblas_fill uplo,
|
|
rocblas_int n,
|
|
rocblas_int nb,
|
|
rocblas_int* kb,
|
|
T* A,
|
|
rocblas_int lda,
|
|
rocblas_int* ipiv,
|
|
T* W,
|
|
rocblas_int ldw,
|
|
rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_sytf2(rocblas_fill uplo,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
rocblas_int* ipiv,
|
|
rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_sytrf(rocblas_fill uplo,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
rocblas_int* ipiv,
|
|
T* work,
|
|
rocblas_int lwork,
|
|
rocblas_int* info);
|
|
|
|
template <typename T>
|
|
void cpu_bdsvdx(rocblas_fill uplo,
|
|
rocblas_svect svect,
|
|
rocblas_srange srange,
|
|
rocblas_int n,
|
|
T* D,
|
|
T* E,
|
|
T vl,
|
|
T vu,
|
|
rocblas_int il,
|
|
rocblas_int iu,
|
|
rocblas_int* nsv,
|
|
T* S,
|
|
T* Z,
|
|
rocblas_int ldz,
|
|
T* work,
|
|
rocblas_int* iwork,
|
|
rocblas_int* info);
|
|
|
|
template <typename T, typename W>
|
|
void cpu_gesvdx(rocblas_svect leftv,
|
|
rocblas_svect rightv,
|
|
rocblas_srange srange,
|
|
rocblas_int m,
|
|
rocblas_int n,
|
|
T* A,
|
|
rocblas_int lda,
|
|
W vl,
|
|
W vu,
|
|
rocblas_int il,
|
|
rocblas_int iu,
|
|
rocblas_int* nsv,
|
|
W* S,
|
|
T* U,
|
|
rocblas_int ldu,
|
|
T* V,
|
|
rocblas_int ldv,
|
|
T* work,
|
|
rocblas_int lwork,
|
|
W* rwork,
|
|
rocblas_int* iwork,
|
|
rocblas_int* info);
|