
oneDNN
uxlfoundation
oneAPI Deep Neural Network Library (oneDNN)
AI 简介
oneDNN 是一个开源的跨平台深度学习性能优化库,提供CPU和GPU上常用神经网络算子(如卷积、池化、归一化、激活函数等)的高度优化实现。它基于oneAPI规范设计,支持x86-64、AArch64及Intel GPU等多种架构,集成AVX-512、AMX、VNNI、BF16等硬件加速特性,并兼容OpenMP、TBB、SYCL等并行编程模型。主要面向深度学习框架开发者,用于提升推理与训练在主流处理器上的执行效率,适用于AI推理引擎、边缘部署、高性能训练优化等场景。
C++
4k
Stars
1.2k
Forks
164
Watchers
30
Issues
Star 增长
今日0
近 7 天0
近 30 天+3
综合评分58.5
默认分支main