UCSB开源TPU项目:OpenTPU横空出世
2025-05-28
加州大学圣巴巴拉分校的ArchLab实验室发布了OpenTPU,这是一个谷歌Tensor Processing Unit (TPU)的开源重新实现。OpenTPU基于谷歌的论文,但由于缺乏正式规范,其设计与谷歌TPU存在差异。目前,OpenTPU支持矩阵乘法和ReLU、sigmoid激活函数,但缺少卷积、池化等功能。该项目使用PyRTL实现,并提供硬件和功能模拟器,以及用于验证结果的检查器。虽然OpenTPU目前尚不具备完整的256x256规模的硬件合成数据,但其开源性质为研究人员提供了宝贵的学习和改进机会。
硬件