batchlas::sycl_potrf Namespace Reference

Functions

potrf_tiny_max_n()

template<typename T>

BATCHLAS_INTERNAL_API int batchlas::sycl_potrf::potrf_tiny_max_n

(

)

potrf_tiny_buffer_size()

template<typename T>

BATCHLAS_INTERNAL_API std::size_t batchlas::sycl_potrf::potrf_tiny_buffer_size	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	A )

potrf_tiny_debug_launch()

template<typename T>

BATCHLAS_INTERNAL_API unsigned batchlas::sycl_potrf::potrf_tiny_debug_launch	(	Queue &	ctx,
		int	n )

potrf_tiny_dispatch()

template<typename T>

BATCHLAS_INTERNAL_API Event batchlas::sycl_potrf::potrf_tiny_dispatch	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	A,
		Uplo	uplo,
		Span< std::byte >	workspace,
		Span< int32_t >	info )

potrf_cta_max_n_for_slm()

template<typename T>

BATCHLAS_INTERNAL_API int batchlas::sycl_potrf::potrf_cta_max_n_for_slm	(	std::size_t	slm_budget_bytes,
		int	min_blocks_per_sm = resident::kMinBlocksPerSm )

potrf_cta_max_n()

template<typename T>

int batchlas::sycl_potrf::potrf_cta_max_n

(

)

potrf_lpanel_max_n_for_slm()

template<typename T>

BATCHLAS_INTERNAL_API int batchlas::sycl_potrf::potrf_lpanel_max_n_for_slm	(	std::size_t	slm_budget_bytes,
		int	max_wg_size,
		int	min_blocks_per_sm = resident::kMinBlocksPerSm,
		int	nb_hint = 0 )

potrf_lpanel_buffer_size()

template<typename T>

BATCHLAS_INTERNAL_API std::size_t batchlas::sycl_potrf::potrf_lpanel_buffer_size	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	A )

potrf_lpanel_debug_launch()

template<typename T>

BATCHLAS_INTERNAL_API unsigned batchlas::sycl_potrf::potrf_lpanel_debug_launch	(	Queue &	ctx,
		int	n,
		int	batch,
		int	min_blocks_per_sm = resident::kMinBlocksPerSm,
		int	nb_hint = 0 )

potrf_lpanel_dispatch()

template<typename T>

BATCHLAS_INTERNAL_API Event batchlas::sycl_potrf::potrf_lpanel_dispatch	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	A,
		Uplo	uplo,
		Span< std::byte >	workspace,
		Span< int32_t >	info,
		int	min_blocks_per_sm = resident::kMinBlocksPerSm,
		int	nb_hint = 0 )

potrf_blocked_available()

template<typename T>

BATCHLAS_INTERNAL_API bool batchlas::sycl_potrf::potrf_blocked_available

(

)

potrf_cta_buffer_size()

template<typename T>

BATCHLAS_INTERNAL_API std::size_t batchlas::sycl_potrf::potrf_cta_buffer_size	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	A )

potrf_cta_debug_launch()

template<typename T>

BATCHLAS_INTERNAL_API unsigned batchlas::sycl_potrf::potrf_cta_debug_launch	(	Queue &	ctx,
		int	n,
		int	batch,
		int	min_blocks_per_sm = resident::kMinBlocksPerSm )

potrf_cta_dispatch()

template<typename T>

BATCHLAS_INTERNAL_API Event batchlas::sycl_potrf::potrf_cta_dispatch	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	A,
		Uplo	uplo,
		Span< std::byte >	workspace,
		Span< int32_t >	info,
		int	min_blocks_per_sm = resident::kMinBlocksPerSm )

potrf_blocked_buffer_size()

template<typename T>

BATCHLAS_INTERNAL_API std::size_t batchlas::sycl_potrf::potrf_blocked_buffer_size	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	A,
		Uplo	uplo )

potrf_blocked_debug_params()

template<typename T>

BATCHLAS_INTERNAL_API unsigned batchlas::sycl_potrf::potrf_blocked_debug_params	(	Queue &	ctx,
		int	n )

potrf_blocked_dispatch()

template<typename T>

BATCHLAS_INTERNAL_API Event batchlas::sycl_potrf::potrf_blocked_dispatch	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	A,
		Uplo	uplo,
		Span< std::byte >	workspace,
		Span< int32_t >	info,
		PotrfTrailingGemm< T >	trailing_gemm,
		PotrfPanelSolve< T >	panel_solve )

Types

PotrfTrailingGemm

template<typename T>

using batchlas::sycl_potrf::PotrfTrailingGemm

Initial value:

std::function<Event(
   Queue&,
   const MatrixView<T, MatrixFormat::Dense>&,
   const MatrixView<T, MatrixFormat::Dense>&,
   const MatrixView<T, MatrixFormat::Dense>&,
   T, T, Transpose, Transpose, ComputePrecision)>

Definition at line 109 of file potrf_native.hh.

PotrfPanelSolve

template<typename T>

using batchlas::sycl_potrf::PotrfPanelSolve

Initial value:

std::function<Event(
   Queue&,
   const MatrixView<T, MatrixFormat::Dense>&,   
   const MatrixView<T, MatrixFormat::Dense>&,   
   T,                                           
   Side, Uplo, Transpose, Diag)>

Definition at line 118 of file potrf_native.hh.

Variables

kPotrfTinyWgSize

int batchlas::sycl_potrf::kPotrfTinyWgSize = 64

inlineconstexpr

Definition at line 25 of file potrf_native.hh.