muDNN

MUSA

muDNN是基于MUSA开发的深度神经网络的计算加速库。在摩尔线程GPU上进行深度优化。 为方便开发者,muDNN软件包现在随MUSA Toolkits一起发布,请移步MUSA Toolkits下载。

MUSA Toolkit

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

MUSA Toolkit

SDK特性

支持Conv,Matmul, 和NHWC,NCHW输入等
支持多数据类型,FP32,FP16,INT8等
加速多种CNN,Transformer,Diffusion模型等

muDNN加速框架