SOLAR:基於AI的光速效能分析框架
SOLAR是一個自動化框架,能夠從PyTorch和JAX原始碼中推匯出經過驗證的光速(SOL)效能界限。它結合了生成式大語言模型和確定性元件,提供未融合、融合及快取感知的SOL界限,並支援多保真度分析。在KernelBench、JAX/Flax模型和機器人工作負載上的實驗展示了其四項用例:多級淨空分析、識別最佳化機會、跨平臺探索以及逆屋頂線硬體配置。
SOLAR:基於AI的光速效能分析框架
本文僅提供摘要
根據來源內容權限,本站僅提供編輯摘要、要點和原文連結。