提速达2.61倍
一个面向脉冲神经网络学习算法的多层级开发与计算框架,兼容PyTorch生态并优化定制算子调度。
Meng, Pang · Yuchen, Li · 李兆麟 · 张悠慧
2023
FABLE采用同步数据流计算模型,能够表达SNN的时空并行性,更好地组织和调度底层算子构建优化计算流程。
灵活的编程接口允许用户设计或定制自己的学习算法,降低了实现复杂度。
扩展了PyTorch的算子库、调度器和API,让用户能够利用PyTorch生态和易用性。
实验表明,FABLE的计算性能超过了所有原始实现,最高加速达2.61倍。