potrs_batch (Group Version)¶
Solves a batch of systems of linear equations with a Cholesky-factored
symmetric (Hermitian) positive-definite coefficient matrices. This
routine belongs to the oneapi::mkl::lapack
namespace.
Description¶
The routine solves for X
i the systems of linear
equations A
i*X
i = B
i with
a symmetric positive-definite or, for complex data, Hermitian
positive-definite matrices A
i, given the Cholesky
factorization of A
i, i ϵ{1...batch_size}
:
A
i =U
iT*U
i for real data,A
i =U
iH*U
i for complex data if uplog=mkl::uplo::upper
A
i =L
i*L
iT for real data,A
i =L
i*L
iH for complex data if uplog=mkl::uplo::lower
where L
i is a lower triangular matrix and
U
i is upper triangular, g
is an index of group of
parameters corresponding to A
i, and the total number of
problems to solve, batch_size
, is a sum of sizes for all of the
groups of parameters as provided by the group_sizes
array.
The systems are solved with multiple right-hand sides stored in the
columns of the matrices B
i.
Before calling this routine, matrices A
i should be
factorized by a call to potrf_batch (Group Version).
API¶
Syntax¶
namespace oneapi::mkl::lapack {
cl::sycl::event potrs_batch(cl::sycl::queue &queue,
mkl::uplo *uplo, std::int64_t *n,
std::int64_t *nrhs,
T **a,
std::int64_t *lda,
T **b,
std::int64_t *ldb,
std::int64_t group_count,
std::int64_t *group_sizes,
T *scratchpad,
std::int64_t scratchpad_size,
const std::vector<cl::sycl::event> &events = {})
}
Function supports the following precisions and devices.
T |
Devices supported |
---|---|
|
Host, CPU, and GPU |
|
Host, CPU, and GPU |
|
Host, CPU, and GPU |
|
Host, CPU, and GPU |
Input Parameters¶
- queue
Device queue where calculations will be performed.
- uplo
Array of
group_count
uplog parameters.Each of uplog indicates whether the upper or lower triangular parts of the input matrices are provided:
If uplog=
mkl::uplo::upper
, input matrices from arraya
belonging to groupg
store the upper triangular parts.If uplog=
mkl::uplo::lower
, input matrices from arraya
belonging to groupg
store the lower triangular parts.- n
Array of
group_count
parametersn
g.Each
n
g specifies the order of the input matrices from arraya
belonging to groupg
.- nrhs
Array of
group_count
parameters nrhsg parameters.Each nrhsg specifies the number of right-hand sides supplied for group
g
in corresponding part of arrayb
.- a
Array of
batch_size
pointers to Cholesky factored matricesA
i as returned by potrf_batch (Group Version).- lda
Array of
group_count
parameters ldag.Each ldag specifies the leading dimension of matrices from
a
belonging to groupg
.- b
Array of
batch_size
pointers to right-hand side matricesB
i, each of size ldbg*nrhsg, whereg
is an index of group corresponding toB
i.- ldb
Array of
group_count
parameters ldbg.Each ldbg specifies the leading dimension of matrices from
b
belonging to groupg
.- group_count
Specifies the number of groups of parameters. Must be at least 0.
- group_sizes
Array of group_count integers. Array element with index
g
specifies the number of problems to solve for each of the groups of parametersg
. So the total number of problems to solve,batch_size
, is a sum of all parameter group sizes.- scratchpad
Scratchpad memory to be used by routine for storing intermediate results.
- scratchpad_size
Size of scratchpad memory as a number of floating point elements of type T. Size should not be less then the value returned by potrs_batch_scratchpad_size (Group Version).
- events
List of events to wait for before starting computation. Defaults to empty list.
Output Parameters¶
- b
The matrices pointed to by array b are overwritten by the solution matrices
X
i.
Exceptions¶
Exception |
Description |
---|---|
|
This exception is thrown when problems occur during calculations. You can obtain the info code of the problem using the info() method of the exception object: If If If |
Return Values¶
Output event to wait on to ensure computation is complete.