FABLE计算框架

提速达2.61倍

一个面向脉冲神经网络学习算法的多层级开发与计算框架,兼容PyTorch生态并优化定制算子调度。

Meng, Pang · Yuchen, Li · 李兆麟 · 张悠慧

2023

参数信息

计算性能加速比
2.61 ×

技术优势

表达时空并行性

FABLE采用同步数据流计算模型,能够表达SNN的时空并行性,更好地组织和调度底层算子构建优化计算流程。

简化算法开发

灵活的编程接口允许用户设计或定制自己的学习算法,降低了实现复杂度。

无缝兼容PyTorch

扩展了PyTorch的算子库、调度器和API,让用户能够利用PyTorch生态和易用性。

大幅提升性能

实验表明,FABLE的计算性能超过了所有原始实现,最高加速达2.61倍。

应用场景