该工具可以预测代码在芯片上运行的速度

麻省理工学院的研究人员发明了一种机器学习工具，可以预测计算机芯片执行来自各种应用程序的代码的速度。  

为了使代码尽可能快地运行，开发人员和编译器（将编程语言转换为机器可读代码的程序）通常使用性能模型，这些模型通过模拟给定的芯片X结构来运行代码。

编译器使用该信息来自动优化代码，而开发人员使用该信息来解决将运行该代码的微处理器的性能瓶颈。但是，机器代码的性能模型是由相对较少的专家小组手写的，并且没有得到正确的验证。结果，模拟的性能测量通常会偏离实际结果。

在一系列会议论文中，研究人员描述了一种新颖的机器学习管道，该管道可自动执行此过程，从而使其变得更轻松，更快，更准确。在 6月份的国际机器学习大会上发表的  一篇论文中，研究人员介绍了Ithemal，这是一种神经网络模型，它以“基本块”（计算指令的基本摘要）形式训练标记的数据，以自动预测其持续时间使用给定的芯片来执行以前看不见的基本块。结果表明，Ithemal的性能比传统的手动调整模型要准确得多。

然后，在11月举行的IEEE国际工作负载特征研讨会上，研究人员介绍  了一个基准域套件，这些  套件来自多个领域，包括机器学习，编译器，密码学和可用于验证性能模型的图形。他们将超过300,000个已分析的块合并到一个名为BHive的开源数据集中。在评估过程中，Ithemal预测了英特尔芯片运行代码的速度要比英特尔自身构建的性能模型还要好。

最终，开发人员和编译器可以使用该工具生成在越来越多的多样化“黑匣子”芯片设计上运行得更快，更高效的代码。现代计算机处理器不透明，极其复杂且难以理解。编写对这些处理器尽可能快地执行的计算机代码也面临着巨大的挑战。计算机科学与人工智能实验室（CSAIL）。“该工具是朝着对这些芯片的性能进行完全建模以提高效率的重要一步。”

最近，在 12月NeurIPS会议上发表的  一篇论文中，该团队提出了一种自动生成编译器优化的新技术。具体来说，它们会自动生成一个名为Vemal的算法，该算法将某些代码转换为向量，可用于并行计算。Vemal优于LLVM编译器中使用的手工矢量化算法，LLVM编译器是业界常用的编译器。

从数据中学习

卡宾说，手工设计性能模型可能是“一门妖术”。英特尔提供了超过3000页的详尽文档，描述了其芯片的X结构。但是，目前只有一小部分专家将构建性能模型，以模拟这些架构上代码的执行。

EECS和CSAIL的研究生Charith Mendis是三篇论文的合著者之一，他说：“英特尔的文档既没有错误也不完整，并且英特尔会省略某些事情，因为它是专有的。” “但是，当您使用数据时，您不需要了解文档。如果有隐藏的东西，您可以直接从数据中学习。”

为此，研究人员记录了给定微处理器用于计算基本块指令的平均周期数-基本上是启动，执行和关闭的顺序-无需人工干预。使过程自动化可以快速分析数十万或数百万个块。

特定领域的架构

在训练中，Ithemal模型分析了数百万个自动剖析的基本块，以准确了解不同的芯片X结构将如何执行计算。重要的是，Ithemal将原始文本作为输入，不需要手动向输入数据添加功能。在测试中，Ithemal可以被喂入以前看不见的基本块和给定的芯片，并且将生成一个数字，指示芯片将以多快的速度执行该代码。

研究人员发现，Ithemal的准确性降低了错误率，这意味着预测速度与实际速度之间的差额比传统的手工模型降低了50％。此外，在他们的下一篇论文中，他们表明Ithemal的错误率为10％，而Intel性能预测模型在多个不同领域的各种基本块上的错误率为20％。

Mendis说，该工具现在使人们可以更轻松地快速了解任何新芯片架构的性能速度。例如，特定领域的X结构，例如Gооgle专门用于神经网络的新型Tensor处理单元，目前正在构建中，但并未得到广泛理解。Mendis说：“如果要在某种新架构上训练模型，只需从该架构收集更多数据，通过我们的探查器运行它，使用该信息来训练Ithemal，现在您就有了一个预测性能的模型，” Mendis说。

接下来，研究人员正在研究使模型可解释的方法。机器学习在很大程度上是一个黑匣子，因此尚不清楚为什么特定模型会做出预测。“我们的模型说一个处理器要执行一个基本块需要10个周期。现在，我们正在尝试找出原因。” Carbin说。“对于这些类型的工具而言，这是一个很好的粒度级别，将是惊人的。”

他们还希望使用Ithemal进一步增强Vemal的性能，并自动实现更好的性能。