AMDとIntel、x86 CPU向けPyTorch最適化を統合へ

　x86 CPUでPyTorchを動かすための最適化ライブラリが、AMDとIntelで共通化される。AMDとIntelは10月20日(現地時間)、それぞれ進めてきたx86 CPU向けのPyTorch最適化を協力して「oneDNN」に統合すると発表した。

　oneDNNは、AIワークロードをCPU上で効率よく実行するためのオープンソースのディープラーニングライブラリで、UXL Foundation(Unified Acceleration Foundation)が管理している。統合により、開発者はAMDとIntelのどちらのCPUでも同じソフトウェアを通じて最適化を利用できるようになり、AIアプリケーションの開発や保守、展開が容易になるとしている。

　AMDは自社CPU向けのライブラリ「ZenDNN」の最適化をoneDNNへ組み込むとともに、UXL Foundationにステアリングメンバーとして加盟した。統合後もAMDプロセッサ向けの最適化は継続するという。

　また両社は、x86の次世代命令セットアーキテクチャ(ISA)「AI Compute Extensions」(ACE)でも協力している。ACEはAIワークロードの中核となる行列乗算を高速化するための拡張で、今後のoneDNNのアップデートを通じてPyTorchに統合される予定。