muDNN

DNN
AI

muDNN是基于MUSA开发的深度神经网络的计算加速库。在摩尔线程GPU上进行深度优化。应用于AI计算加速,广泛集成与PyTorch,Paddle等AI框架之中。

MUSA Toolkit

MUSA Toolkit包含了运行时、编译器、GPU加速计算库,与迁移和优化工具。助开发者高效地开发与迁移GPU应用程序。

MUSA Toolkit

SDK 特性

支持Conv,Matmul, 和NHWC,NCHW输入等
支持多数据类型,FP32,FP16,INT8等
加速多种CNN,Transformer,Diffusion模型等
与主流AI框架集成

muDNN加速AI框架