研究人员开发了TIRx,这是一种新的机器学习编译器,旨在优化跨GPU和AI加速器等多样化硬件的性能。TIRx引入了显式的执行范围抽象,并使用命名轴来表示张量布局,将逻辑张量映射到异构硬件空间。这种方法旨在提供原生级别的控制,同时自动化复杂的代码生成,据报道在各种硬件上实现了与供应商同等水平的性能。 AI
影响 旨在提高机器学习模型在多样化硬件平台上的性能和效率。
排序理由 关于新型机器学习编译器的学术论文。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →