potrf_batch (USM ストライドバージョン)

一連の対称 (または複素データの場合はエルミート) 正定値行列のコレスキー分解を計算します。このルーチンは、oneapi::mkl::lapack 名前空間に属します。

説明

このルーチンは、対称正定値行列、または複素数データの場合はエルミート正定値行列 A_i, iϵ{1...batch_size} のコレスキー分解を形成します。

• 実データの場合は A_i = U_i^T * U_i、複素データの場合は A_i = U_i^H * U_i。 uplog = mkl::uplo::upper の場合、

• 実データの場合は A_i = L_i^T * L_i、複素データの場合は A_i = L_i^H * L_i。uplo = mkl::uplo::lower の場合

ここで、L_i は下三角行列、U_i は上三角行列です。

API

構文

namespace oneapi::mkl::lapack { 
  sycl::event potrf_batch(sycl::queue &queue, 
  mkl::uplo uplo, 
  int64_t n, 
  T *a, int64_t lda, 
  int64_t stride_a, 
  int64_t batch_size, 
  T *scratchpad, 
  int64_t scratchpad_size, 
  const std::vector<sycl::event> &events = {}) 
}

この関数は次の精度とデバイスをサポートしています。

T	デバイスサポート
float	CPU および GPU
double	CPU および GPU
std::complex<float>	CPU および GPU
std::complex<double>	CPU および GPU

入力パラメーター

queue

計算が実行されるデバイスキュー。

uplo

A_i の上三角部分と下三角部分のどちらが格納されているか、また A_i がどのように因数分解されるかを示します。

uplo = uplo::upper の場合、配列 a には行列 A_i の上三角部分が格納されます。

uplo = uplo::lower の場合、配列 a には行列 A_i の下三角部分が格納されます。

n

行列 A_i の次数を指定します (n ≥ 0)。

a

入力行列 A_i のバッチを含む配列。各 A_i はサイズ lda*n であり、行列 A_i の上三角部分または下三角部分のいずれかを保持します (uplo を参照)。

lda

A_i の先頭次元 i(lda ≥ max(1, n))。

stride_a

バッチ内の行列 A_i の先頭間のストライド (stride_a ≥ max(1, lda * n))。

batch_size

バッチ内の問題数 (batch_size ≥ 0)。

scratchpad

ルーチンが中間結果を保存するスクラッチパッド・メモリー。

scratchpad_size

T タイプの浮動小数点要素数のスクラッチパッド・メモリーのサイズ。サイズは、potrf_batch_scratchpad_size (ストライドバージョン) が返す値よりも小さくてはなりません。

events

計算を開始する前に待機するイベントのリスト。デフォルトでは空のリストになります。

出力パラメーター

a

バッチ配列 a は、uplo で指定されたコレスキー係数 U_i または L_i によって上書きされます。

例外

例外	説明
mkl::lapack::batch_exception	この例外は、計算中に問題が発生した場合にスローされます。例外オブジェクトの info() メソッドを使用して、問題の情報コードを取得できます。 info = -n の場合、n 番目のパラメーターの値が不正です。 info がスクラッチパッドのサイズとして渡された値と等しく、detail() がゼロ以外を返す場合、渡されたスクラッチパッドのサイズが十分ではなく、必要なサイズは例外オブジェクトの detail() メソッドによって返される値以上である必要があります。 info がゼロの場合、Ui の一部の対角要素がゼロになり、解決を完了できません。バッチ内のこのような行列のインデックスは、例外オブジェクトの ids() メソッドを使用して取得できます。例外オブジェクトの infos() メソッドを使用して、これらの Ui 行列の最初のゼロ対角要素のインデックスを取得できます。

戻り値

計算が完了したことを確認するために待機する出力イベント。