batchlas::internal Namespace Reference

Functions

givens_rotation()

template<typename T>

std::array< T, 2 > batchlas::internal::givens_rotation ( const T & a, const T & b )

inline

Definition at line 15 of file steqr_internal.hh.

apply_givens_rotation()

template<typename T>

T batchlas::internal::apply_givens_rotation ( const VectorView< T > & d, const VectorView< T > & e, const T & prev_bulge, size_t i, size_t j, const std::array< T, 2 > & givens, bool QR )

inline

Definition at line 25 of file steqr_internal.hh.

make_strided() [1/2]

template<typename T>

StridedView< T > batchlas::internal::make_strided ( T * ptr, int inc )

inline

Definition at line 34 of file sytrd_sb2st_cta.hh.

make_strided() [2/2]

template<typename T>

ConstStridedView< T > batchlas::internal::make_strided ( const T * ptr, int inc )

inline

Definition at line 39 of file sytrd_sb2st_cta.hh.

make_const_strided()

template<typename T>

ConstStridedView< T > batchlas::internal::make_const_strided ( T * ptr, int inc )

inline

Definition at line 44 of file sytrd_sb2st_cta.hh.

lartv_impl()

template<typename Ops, typename T>

void batchlas::internal::lartv_impl ( int nrot, StridedView< T > x, StridedView< T > y, ConstStridedView< typename Ops::Real > c, ConstStridedView< T > s )

inline

Definition at line 49 of file sytrd_sb2st_cta.hh.

largv_impl()

template<typename Ops, typename T>

void batchlas::internal::largv_impl ( int nrot, StridedView< T > x, StridedView< T > w, StridedView< typename Ops::Real > c )

inline

Definition at line 57 of file sytrd_sb2st_cta.hh.

real_as_T()

template<typename T>

T batchlas::internal::real_as_T ( const T & x )

inline

Definition at line 72 of file sytrd_sb2st_cta.hh.

btrd_lower_inplace_subgroup_dispatch()

template<typename T>

Event batchlas::internal::btrd_lower_inplace_subgroup_dispatch	(	Queue &	q,
		KernelMatrixView< T, MatrixFormat::Dense >	ab,
		int	n,
		int	kd,
		VectorView< typename base_type< T >::type >	c,
		VectorView< T >	work,
		VectorView< typename base_type< T >::type >	d,
		VectorView< typename base_type< T >::type >	e,
		VectorView< T >	tau )

build_sb2st_hh_schedule()

std::vector< Sb2stHhRefl > batchlas::internal::build_sb2st_hh_schedule ( int32_t n, int32_t kd )

inline

Definition at line 29 of file sytrd_sb2st_hh.hh.

sb2st_hh_num_reflectors()

int32_t batchlas::internal::sb2st_hh_num_reflectors ( int32_t n, int32_t kd )

inline

Definition at line 52 of file sytrd_sb2st_hh.hh.

sb2st_hh_work_bandwidth()

int32_t batchlas::internal::sb2st_hh_work_bandwidth ( int32_t n, int32_t kd )

inline

Definition at line 57 of file sytrd_sb2st_hh.hh.

sytrd_sb2st_hh()

template<Backend B, typename T>

BATCHLAS_INTERNAL_API Event batchlas::internal::sytrd_sb2st_hh	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	ab_in,
		const MatrixView< T, MatrixFormat::Dense > &	ab_tri_out,
		const VectorView< typename base_type< T >::type > &	d_out,
		const VectorView< typename base_type< T >::type > &	e_out,
		const MatrixView< T, MatrixFormat::Dense > &	v_out,
		const VectorView< T > &	tau_out,
		Uplo	uplo,
		int32_t	kd,
		const Span< std::byte > &	ws )

sytrd_sb2st_hh_buffer_size()

template<Backend B, typename T>

BATCHLAS_INTERNAL_API size_t batchlas::internal::sytrd_sb2st_hh_buffer_size	(	Queue &	ctx,
		int32_t	n,
		int32_t	kd,
		int32_t	batch )

build_sb2st_hh_wave_offsets()

std::vector< int32_t > batchlas::internal::build_sb2st_hh_wave_offsets ( const std::vector< Sb2stHhRefl > & sched, int32_t n )

inline

Definition at line 84 of file sytrd_sb2st_hh.hh.

unmqr_hb2st()

template<Backend B, typename T>

BATCHLAS_INTERNAL_API Event batchlas::internal::unmqr_hb2st	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	v_in,
		const VectorView< T > &	tau_in,
		const MatrixView< T, MatrixFormat::Dense > &	z_io,
		int32_t	n,
		int32_t	kd,
		Span< const int32_t >	starts,
		Span< const int32_t >	lens,
		Span< const int32_t >	waves )

Classes
struct	btrd_givens_ops
struct	btrd_givens_ops< T, false >
struct	btrd_givens_ops< T, true >
struct	ConstStridedView
struct	Sb2stHhRefl
struct	StridedView