AMDとIntel、PyTorchのAI高速化にoneDNNを選定
AMDとIntelは、x86アーキテクチャのプロセッサー上で人工知能のタスクを処理するためのフレームワークPyTorchの最適化について、取り組みを統合することを決定しました。共通の基盤となるのは、オープンソースのディープラーニングライブラリoneDNNです。このライブラリは、計算高速化のためのオープンな技術を推進する財団Unified Acceleration Foundation（UXL）の支援を受けて開発されています。
今回の合意は、諮問グループx86 Ecosystem Advisory Groupの勧告を踏まえて成立しました。合意に基づき、oneDNNはx86プロセッサー上のPyTorchにおけるAIのハードウェア高速化を担う主要なオープンソースライブラリとして採用されます。これにより開発者は、各社のハードウェアごとに分かれた最適化の仕組みではなく、統一されたソフトウェア基盤を利用できるようになります。
報道によると、AMDはすでに自社ライブラリZenDNNの最適化をoneDNNのコードベースへ移行し始めています。また同社は、UXL Foundationの運営委員会にも加わりました。業界関係者は、この動きによってx86のソフトウェアエコシステムの分断が緩和されると期待しています。
今後の共同作業の次の焦点は、AI Compute Extensionsの開発です。これは、ニューラルネットワークの学習や実行で広く使われる行列乗算を高速化するための、x86命令セットの拡張群です。新しい命令のサポートは、oneDNNのアップデートを通じてPyTorchに組み込まれる予定です。
ただし、この技術は現在も開発段階にあります。公式発表では、拡張機能がいつ完成するのか、またいつ量産プロセッサーに搭載されるのかについては明らかにされていません。
Intelは、統一されたソフトウェア経路により、開発者がプロセッサーの製造元に応じて異なる最適化ツールを選ぶ必要がなくなると説明しました。AMDは、この協力がオープンなソフトウェアエコシステムの発展に寄与し、高性能なAIアプリケーションの開発を容易にすると強調しました。PyTorchコミュニティに参加しているRed Hatの代表者も、ツールの統一を支持しています。
なお、今回の合意が主に対象とするのはソフトウェア基盤とその今後の発展であり、新型プロセッサーや完成したアクセラレーターの発売ではありません。共通ライブラリは、AMDとIntelが既に備えている機能の利用を容易にし、AI Compute Extensionsの登場後には、新しいハードウェア機能をPyTorchに接続するための統一的な手段を提供するとされています。
