GPU Forecasters: Language Models as Selective Surrogates for Kernel Runtime Optimization
GPU预测器:语言模型作为内核运行时优化的选择性替代
机构 * UNC Chapel Hill(北卡罗来纳大学教堂山分校) ; AI2 ; Johns Hopkins University(约翰霍普金斯大学) ; University of Texas at Austin(德克萨斯大学奥斯汀分校)
AI总结 研究利用语言模型作为GPU内核性能的选择性替代,通过强化学习提高预测准确性和校准度,在有限GPU评估预算下加速内核搜索。
Comments Code: https://github.com/codezakh/gpu-forecasters