Sep, 2023

基于贝叶斯优化的自动调整 Apache TVM 科学应用

TL;DR本论文通过使用贝叶斯优化提出了一种新的 TVM 自动调优框架,并利用 TVM 张量表达语言实现了 LU、Cholesky 和 3mm 等线性代数核,通过在 Argonne 国家实验室的一个 GPU 集群 Swing 上评估这些方法的有效性,将所提出的自动调优框架与四个调优器的 TVM 自动调优框架 AutoTVM 进行了比较,并发现在大多数情况下我们的框架优于 AutoTVM。