arXiv:2406.17811v2 宣布类型: 替换-交叉 摘要：贝叶斯优化是用于自动化编译器调优的强大方法。自动调优的复杂景观为黑盒优化器提供了众多鲜有考虑的结构性挑战，而标准化基准的缺乏限制了贝叶斯优化在该领域的研究。为了解决这一问题，我们提出了CATBench，这是一个全面的基准测试套件，能够捕捉编译器自动调优的复杂性，包括从离散、条件和排列参数类型到已知和未知的二进制约束，以及多保真度和多目标评估。CATBench中的基准测试涵盖了从张量代数到图像处理和聚类等一系列机器学习导向的计算任务，并使用了最先进的编译器，如TACO和RISE/ELEVATE。CATBench提供了一种统一的接口来评估贝叶斯优化算法，并通过易于使用的完全容器化设置促进编译器优化任务中的重复性和创新性。我们对CATBench进行了多项最先进的算法验证，揭示了这些算法的优势和弱点，并展示了该套件在推动贝叶斯优化和编译器自动调优研究方面的潜力。