互联网资讯 / 人工智能 · 2023年12月11日 0

下一代TPU实现AI自主设计

设计一款AI芯片究竟有多复杂?

可以这样理解,围棋的复杂性为10的360次方,而芯片的复杂性则高达10的2500次方,让人不禁感到震撼。

通常情况下,工程师们在设计一块芯片时,所需的时间从几周到几个月不等。

然而,AI的出现改变了这一切!

AI竟然能够在短短6小时内完成芯片设计。

这一突破最近被谷歌的研究团队发表在《Nature》杂志上。

一块小小的芯片包含数十亿个晶体管,它们形成数千万个逻辑门以及数千个称为宏块的存储块。

确定这些组件的位置,即布局规划,是芯片设计的关键步骤。

布局直接影响到布线方式,从而影响芯片的处理速度和电源效率。

然而,仅仅放置宏块这一环节就十分耗时,每次迭代通常需要几天到几周的时间,以确保为标准单元腾出足够的空间。

完成整个布局的过程可能需要几周甚至几个月。

现在,谷歌的研究人员提出了一种新型的芯片布局方法,具有良好的泛化能力。

该方法利用深度强化学习,从已有布局中学习,进而生成新的设计方案,其整体架构如下:

为了加快学习速度,研究人员设计了一种奖励机制,利用线路长度和布线拥堵的近似代价函数进行评估。

具体而言,需将宏块和标准单元映射到一个平面画布上,形成拥有数百万到数十亿节点的“芯片网表”。

随后,AI模型会对功率、性能和面积等指标进行优化,并输出相应的概率分布。