計算ライブラリ リファレンス
English | 简体中文 | 日本語 | Deutsch | Русский
ドキュメント · テンソルとフレームワーク · 中文
計算ライブラリはオペレーションを実装し、ランタイムがデバイス上でオペレーションを実行し、テンソル フレームワークがオペレーションを構成します。ライブラリ名は責任を説明するものであり、同様の名前の CUDA ライブラリと同一の API や完全な機能の同等性を示すものではありません。
ライブラリを選択してください
| ライブラリ | Cargo パッケージ | Rust クレート | 主な操作 |
|---|---|---|---|
| ruBLAS | rublas |
rublas |
行列乗算、ベクトル演算、グループ化された行列乗算、および INT4 |
| ruDNN | ruDNN |
rudnn |
アテンション、畳み込み、プーリング、および MoE |
| ruTENSOR | ruTENSOR |
rutensor |
一般的なテンソル短縮、einsum、リダクション、順列、および要素ごとの演算 |
| ruPRIM | ruPRIM |
ruprim |
リダクション、スキャン、要素ごとの操作、およびインデックス付け |
| ruFFT | ruda-fft |
rufft |
実数 FFT と逆変換 |
| ruRAND | ruRAND |
rurand |
一様分布、正規分布、およびベルヌーイ分布 |
| ruSPARSE | ruSPARSE |
rusparse |
スパース行列の形式と演算 |
| ruCCL | ruCCL |
ruccl |
集団的なコミュニケーションとオーケストレーション |
インターフェイス層
- カーネル/起動インターフェイスは、カーネルおよびライブラリ開発のためのデバイス バインディングと実行構成を受け入れます。
- Tensor インターフェイスは、通常
RudaTensor<R>を使用して、割り当て、レイアウト、呼び出しを処理します。 - フレームワーク インターフェイスは、
ruda-tensor-deviceなどのコンポーネントを通じてライブラリにディスパッチします。
異なるレイヤーで同じ名前の操作では、引数、戻り値の型、およびエラー処理が異なる場合があります。パッケージ、モジュール パス、および機能を一緒に確認します。カーネル エントリ ポイント署名は、テンソル エントリ ポイント署名ではありません。
ガイドの使用方法
各ライブラリ ガイドでは、その目的、機能、インターフェイス、データ コントラクト、および制約について説明します。デフォルトがすべてのデバイスに適合すると仮定するのではなく、汎用パスに必要な機能を明示的に選択します。