ROCm/MIOpen

何を解決するか

MIOpen は AMD GPU 向けの高性能機械学習プリミティブを提供するライブラリです。ROCm 対応プラットフォーム上で効率的な演算を実行するための深層学習フレームワークの基盤となる層として機能します。

動作方法

一般的な機械学習演算向けに最適化されたカーネルのセットを実装しています。HIP および OpenCL（非推奨）のプログラミングモデルをサポートしています。起動遅延を低減するために、カーネルキャッシュを使用し、特定の GPU アーキテクチャ（例：gfx900、gfx1030）向けにオプションで事前コンパイルされたカーネルを提供しています。

対象ユーザー

ニューラルネットワーク演算に AMD ハードウェアアクセラレーションを活用する必要がある機械学習フレームワークやアプリケーションの開発者。

主な特徴

• ハードウェアアクセラレーション：ROCm プラットフォームを介して AMD GPU 向けに最適化されています。
• カーネルキャッシュ：再コンパイルを回避するための永続的プログラムキャッシュを使用しています。
• 事前コンパイルされたカーネル：ニューラルネットワークの初期起動時間を大幅に短縮するためのオプションパッケージを提供しています。
• cuDNN の移行：NVIDIA の cuDNN から MIOpen への移行を支援する専用ガイドを提供しています。