要点
IntelとAMDは、PyTorch向けのCPU最適化を単一の統合ライブラリに統合するために協力しています。 AMDは、自社のZenDNNライブラリをUXL Foundationが管理するオープンソースのoneDNNプロジェクトに移行します。
さらに、AMDはUXL Foundationに運営メンバーとして加入し、年間2万ドルを拠出します。 2024年に設立されたx86エコシステム諮問グループによって推奨されたこの取り組みは、開発者のAI推論を簡素化することを目的としています。 oneDNNバックエンドを利用することで、ソフトウェアは実行時にハードウェア機能を自動的に検出し、最適化されたカーネルを生成できます。
このライブラリは、行列乗算や畳み込みを含む重要な機械学習演算をサポートし、低精度データ型を提供します。 2023年後半にPyTorchに統合されたoneDNNバックエンドは、大幅なパフォーマンス向上を実証しました。
報告されたデータによると、このライブラリはサポートされているハードウェア上で、Float32データ型で1.24倍、BFloat16データ型で2.21倍の性能向上を達成しました。 この取り組みはx86アーキテクチャに焦点を当てていますが、oneDNNはArmやRISC-Vもサポートするクロスプラットフォームライブラリであり続けます。 Google CloudやQualcommなどと共にUXL Foundationの創設メンバーであるIntelは、当初oneDNNを開発し、その後財団に移管しました。
