テンソル並列とは
てんそるへいれつ
1つの層の重みの行列を行や列の方向に分けて複数のGPUに持たせ、部分的な積を求めてから通信でまとめる分散の方法。
この用語の分野: AIハードウェアとソフトウェアスタック
てんそるへいれつ
1つの層の重みの行列を行や列の方向に分けて複数のGPUに持たせ、部分的な積を求めてから通信でまとめる分散の方法。
この用語の分野: AIハードウェアとソフトウェアスタック
テンソル並列では、例えば重みの行列を列の方向に4つに分けて4つのGPUに持たせ、各GPUが同じ入力と自分の担当の列の積を計算します。
1つのGPUのメモリに収まらない大きな層の行列の積を、複数のGPUで分けて計算するテンソル並列の説明として正しいものは…GPU データ並列 パイプライン並列 アテンション AIコンパイラ MFU 演算強度 演算の融合 オールリデュース ONNX