getrs_batch.hpp

#include "./getrs.hpp"
#include "./interface/cxx_interface.hpp"
#include "../basic_array.hpp"
#include "../blas/interface/cxx_interface.hpp"
#include "../blas/tools.hpp"
#include "../concepts.hpp"
#include "../declarations.hpp"
#include "../device.hpp"
#include "../layout_transforms.hpp"
#include "../macros.hpp"
#include "../mem/address_space.hpp"
#include "../traits.hpp"
#include <algorithm>
#include <tuple>
#include <type_traits>
#include <utility>

Detailed Description

Provides a generic interface to batched versions of the LAPACK/cuSOLVER getrs routine.

Definition in file getrs_batch.hpp.

Go to the source code of this file.

Functions
template<BlasArrayOrConj< 3 > A, BlasArrayFor< A, 3 > B, PivotArrayFor< A, 2 > IPIV> requires ((has_F_layout<A> or has_C_layout<A>) and has_F_layout<B> and (not is_conj_array_expr<A> or has_C_layout<A>))
int	nda::lapack::getrs (A const &a, B &&b, IPIV const &ipiv)
	Generic-friendly overload of nda::lapack::getrs for batches stored as 3-dimensional arrays.
template<BlasArrayOrConj< 3 > A, BlasArrayFor< A, 3 > B, PivotArrayFor< A, 2 > IPIV> requires ((has_F_layout<A> or has_C_layout<A>) and has_F_layout<B> and (not is_conj_array_expr<A> or has_C_layout<A>))
int	nda::lapack::getrs_batch (A const &a, B &&b, IPIV const &ipiv)
	Interface to batched versions of the LAPACK/cuSOLVER getrs routine.

Function Documentation

getrs()

template<BlasArrayOrConj< 3 > A, BlasArrayFor< A, 3 > B, PivotArrayFor< A, 2 > IPIV>
requires ((has_F_layout<A> or has_C_layout<A>) and has_F_layout<B> and (not is_conj_array_expr<A> or has_C_layout<A>))

int nda::lapack::getrs	(	A const &	a,
		B &&	b,
		IPIV const &	ipiv )

Generic-friendly overload of nda::lapack::getrs for batches stored as 3-dimensional arrays.

It simply calls nda::lapack::getrs_batch and lets generic code call getrs(...) regardless of whether the input is a single matrix (rank 2) or a batch (rank 3).

Definition at line 138 of file getrs_batch.hpp.