gemmt#
一般行列の行列-行列積を計算しますが、結果行列の上三角部分または下三角部分のみを更新します。
説明
gemmt ルーチンは、スカラー - 行列 - 行列積を計算し、その結果を一般行列のスカラー - 行列積の上位または下位に加算します。操作はのように定義されています。
説明:
op(
X) は、op(X) =X、op(X) =XT、または op(X) =XH のいずれかですalphaとbetaはスカラーですA、B、およびCは行列ですop(
A) はnxk、op(B) はkxn、そしてCはnxnです。
gemmt は次の精度をサポートします。
T |
|---|
|
|
|
|
gemmt (バッファーバージョン)#
構文#
namespace oneapi::mkl::blas::column_major {
void gemmt(sycl::queue &queue,
oneapi::mkl::uplo upper_lower,
oneapi::mkl::transpose transa,
oneapi::mkl::transpose transb,
std::int64_t n,
std::int64_t k,
T alpha,
sycl::buffer<T,1> &a,
std::int64_t lda,
sycl::buffer<T,1> &b,
std::int64_t ldb,
T beta,
sycl::buffer<T,1> &c,
std::int64_t ldc,
compute_mode mode = compute_mode::unset)
}namespace oneapi::mkl::blas::row_major {
void gemmt(sycl::queue &queue,
oneapi::mkl::uplo upper_lower,
oneapi::mkl::transpose transa,
oneapi::mkl::transpose transb,
std::int64_t n,
std::int64_t k,
T alpha,
sycl::buffer<T,1> &a,
std::int64_t lda,
sycl::buffer<T,1> &b,
std::int64_t ldb,
T beta,
sycl::buffer<T,1> &c,
std::int64_t ldc,
compute_mode mode = compute_mode::unset)
}入力パラメーター#
- queue
ルーチンを実行するキュー。
- upper_lower
行列
Cが上三角行列か下三角行列であるかを指定します。詳細はデータタイプを参照してください。- transa
行列
Aに適用される転置演算 op(A) を指定します。詳細はデータタイプを参照してください。- transb
行列
Bに適用される転置演算 op(B) を指定します。詳細はデータタイプを参照してください。- n
行列 op(
A) と行列Cの行数。最小値は 0 です。- k
行列 op(
A) の列数と行列 op(B) の行数。最小値は 0 です。- alpha
行列 - 行列積のスケーリング係数。
- a
入力行列
Aを保持するバッファー。行列ストレージを参照してください。transa=transpose::nontranstransa=transpose::transまたはtrans=transpose::conjtrans列優先
Aはnxkの行列です。配列aのサイズはlda*k以上でなければなりませんAはkxnの行列です。配列aのサイズはlda*n以上でなければなりません行優先
Aはnxkの行列です。配列aのサイズはlda*n以上でなければなりませんAはkxnの行列です。配列aのサイズはlda*k以上でなければなりません- lda
行列
Aの先頭次元。正である必要があります。transa=transpose::nontranstransa=transpose::transまたはtrans=transpose::conjtrans列優先
n以上である必要がありますk以上である必要があります行優先
k以上である必要がありますn以上である必要があります- b
入力行列
Bを保持するバッファー。行列ストレージを参照してください。transb=transpose::nontranstransb=transpose::transまたはtrans=transpose::conjtrans列優先
Bはkxnの行列です。配列bのサイズはldb*n以上でなければなりませんBはnxkの行列です。配列bのサイズはldb*k以上でなければなりません行優先
Bはkxnの行列です。配列bのサイズはldb*k以上でなければなりませんBはnxkの行列です。配列bのサイズはldb*n以上でなければなりません- ldb
行列
Bの先頭次元。正である必要があります。transb=transpose::nontranstransb=transpose::transまたはtrans=transpose::conjtrans列優先
k以上である必要がありますn以上である必要があります行優先
n以上である必要がありますk以上である必要があります- beta
行列
Cのスケーリング係数。- c
入力/出力行列
Cを保持するバッファー。行列ストレージを参照してください。列優先
Cはmxnの行列です。配列cのサイズはldc*n以上でなければなりません行優先
Cはmxnの行列です。配列cのサイズはldc*m以上でなければなりません- ldc
行列
Cの先頭次元。正である必要があります。列優先
m以上である必要があります行優先
n以上である必要があります- mode
オプション。計算モードの設定。詳細は、計算モードを参照してください。
出力パラメーター#
- c
出力バッファーは、
alpha* op(A)*op(B) +beta*Cの上三角部分または下三角部分で上書きされます。
注
beta = 0 の場合、gemmt を呼び出す前に行列 C を初期化する必要はありません。
gemmt (USM バージョン)#
構文#
namespace oneapi::mkl::blas::column_major {
sycl::event gemmt(sycl::queue &queue,
oneapi::mkl::uplo upper_lower,
oneapi::mkl::transpose transa,
oneapi::mkl::transpose transb,
std::int64_t n,
std::int64_t k,
oneapi::mkl::value_or_pointer<T> alpha,
const T* a,
std::int64_t lda,
const T* b,
std::int64_t ldb,
oneapi::mkl::value_or_pointer<T> beta,
T* c,
std::int64_t ldc,
compute_mode mode = compute_mode::unset,
const std::vector<sycl::event> &dependencies = {})
}namespace oneapi::mkl::blas::row_major {
sycl::event gemmt(sycl::queue &queue,
oneapi::mkl::uplo upper_lower,
oneapi::mkl::transpose transa,
oneapi::mkl::transpose transb,
std::int64_t n,
std::int64_t k,
oneapi::mkl::value_or_pointer<T> alpha,
const T* a,
std::int64_t lda,
const T* b,
std::int64_t ldb,
oneapi::mkl::value_or_pointer<T> beta,
T* c,
std::int64_t ldc,
compute_mode mode = compute_mode::unset,
const std::vector<sycl::event> &dependencies = {})
}入力パラメーター#
- queue
ルーチンを実行するキュー。
- upper_lower
行列
Cが上三角行列か下三角行列であるかを指定します。詳細はデータタイプを参照してください。- transa
行列
Aに適用される転置演算 op(A) を指定します。詳細はデータタイプを参照してください。- transb
行列
Bに適用される転置演算 op(B) を指定します。詳細はデータタイプを参照してください。- n
行列 op(
A) と行列Cの行数。最小値は 0 です。- k
行列 op(
A) の列数と行列 op(B) の行数。最小値は 0 です。- alpha
行列 - 行列積のスケーリング係数。
value_or_pointerデータタイプの詳細については、スカラー引数を参照してください。- a
入力行列
Aへのポインター。行列ストレージを参照してください。transa=transpose::nontranstransa=transpose::transまたはtrans=transpose::conjtrans列優先
Aはnxkの行列です。配列aのサイズはlda*k以上でなければなりませんAはkxnの行列です。配列aのサイズはlda*n以上でなければなりません行優先
Aはnxkの行列です。配列aのサイズはlda*n以上でなければなりませんAはkxnの行列です。配列aのサイズはlda*k以上でなければなりません- lda
行列
Aの先頭次元。正である必要があります。transa=transpose::nontranstransa=transpose::transまたはtrans=transpose::conjtrans列優先
n以上である必要がありますk以上である必要があります行優先
k以上である必要がありますn以上である必要があります- b
入力行列
Bへのポインター。行列ストレージを参照してください。transb=transpose::nontranstransb=transpose::transまたはtrans=transpose::conjtrans列優先
Bはkxnの行列です。配列bのサイズはldb*n以上でなければなりませんBはnxkの行列です。配列bのサイズはldb*k以上でなければなりません行優先
Bはkxnの行列です。配列bのサイズはldb*k以上でなければなりませんBはnxkの行列です。配列bのサイズはldb*n以上でなければなりません- ldb
行列
Bの先頭次元。正である必要があります。transb=transpose::nontranstransb=transpose::transまたはtrans=transpose::conjtrans列優先
k以上である必要がありますn以上である必要があります行優先
n以上である必要がありますk以上である必要があります- beta
行列
Cのスケーリング係数。value_or_pointerデータタイプの詳細については、スカラー引数を参照してください。- c
入力/出力行列
Cへのポインター。行列ストレージを参照してください。列優先
Cはmxnの行列です。配列cのサイズはldc*n以上でなければなりません行優先
Cはmxnの行列です。配列cのサイズはldc*m以上でなければなりません- ldc
行列
Cの先頭次元。正である必要があります。列優先
m以上である必要があります行優先
n以上である必要があります- mode
オプション。計算モードの設定。詳細は、計算モードを参照してください。
- dependencies
オプション。計算を開始する前に待機するイベントのリスト (存在する場合)。省略した場合、依存関係はデフォルトでなくなります。
modeとdependenciesはそれぞれ省略できます。dependencies を提供するのにmodeを指定する必要はありません。
出力パラメーター#
- c
alpha* op(A)*op(B) +beta*Cの上三角部分または下三角部分で上書きされた出力行列Cへのポインター。
注
beta = 0 の場合、gemmt を呼び出す前に行列 C を初期化する必要はありません。
戻り値#
計算が完了したことを確認するために待機する出力イベント。