github.devotd.
新着推論・サービング

uxlfoundation/onednn v3.13.4 — CPU·GPU 深層学習演算ライブラリ

uxlfoundation/onednn の最新リリース v3.13.4 と、CPU・Intel GPU 向け深層学習演算ライブラリの役割akaran arenas を眺める。

リポジトリ情報uxlfoundation/onednn ↗

Open-source library of optimized deep learning operations (matmul, convolution, attention) for CPUs (x64, AArch64, RISC-V) and Intel GPUs. Used by PyTorch, TensorFlow, OpenVINO, and ONNX Runtime.

スター
★ 4.1K
フォーク
1.2K
言語
C++
ライセンス
Apache-2.0
最新リリース
v3.13.4
原文を見る ↗GitHub で見る ↗

uxlfoundation/onednn の最新リリースは v3.13.4 だ。oneDNN(oneAPI Deep Neural Network Library)は、CPU と Intel GPU 上で行列乗算、畳み込み、アテンションなどの深層学習演算を提供するオープンソースライブラリだ。提供された情報には、このバージョンで変更された項目は別途記載されていない。

このライブラリは x64、AArch64、RISC-V CPU と Intel GPU を対象に最適化されている。README はこれをクロスプラットフォームの深層学習演算ライブラリとして紹介している。行列乗算と畳み込み、アテンション、正規化の最適化実装を提供し、Apache-2.0 ライセンスで配布される。PyTorch、TensorFlow、OpenVINO、ONNX Runtime は oneDNN をアクセラレーション層として利用している。したがって oneDNN はフレームワークを置き換えるというより、フレームワークが呼び出す深層学習演算を実装する基盤に近い。

今回のリリースの具体的な変更点や性能測定値は、提供された情報には載っていない。したがって v3.13.4 が機能や速度をどのように変えたかは、ここで判断できない。確認できるのは、複数の CPU 系と Intel GPU を対象とし、主要なフレームワークで使われている演算ライブラリだという点までだ。