Skip to content

An optimized neural network operator library for chips base on Xuantie CPU.

License

Notifications You must be signed in to change notification settings

zhangwm-pt/csi-nn2

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

简介

SHL(曾用名CSI-NN2) 是 T-HEAD 提供的一组针对玄铁 CPU 平台的神经网络库 API。抽象了各种常用的网络层的接口,并且提供一系列已优化的二进制库。

SHL 的特性:

  • C 代码版本的参考实现。
  • 提供玄铁系列 CPU 的汇编优化实现。
  • 支持对称量化和非对称量化。
  • 支持8位定点,16位定点和16位浮点等数据类型。
  • 兼容 NCHW 和 NHWC 格式。
  • 搭配 HHB 实现代码自动调用。
  • 覆盖 CPU,NPU 等不同体系结构。
  • 附加异构参考实现。

SHL 提供了完成的接口声明和接口的参考实现,各个设备提供商可以依此针对性的完成各个接口的优化工作。

文档说明

致谢

SHL 参考、借鉴了下列项目:

About

An optimized neural network operator library for chips base on Xuantie CPU.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • C 88.3%
  • Assembly 5.7%
  • Python 5.0%
  • C++ 0.7%
  • CMake 0.1%
  • Shell 0.1%
  • Makefile 0.1%