TVM新教程:GPU卷積層自動調度免模板
Apache TVM是適用於CPU、GPU及各類機器學習加速晶片的深度學習編譯框架,本教程講解auto-scheduler如何為GPU自動調度卷積層,毋須手寫模板即可搜索最優調度。
框架定位
TVM面向多種硬件後端,把計算描述編譯為高效實現;對GPU上的卷積層等重點算子,調度方案的好壞直接影響推理與訓練性能,自動調度因此成為關鍵能力。
IXThinker 會員內容
呢篇係 Archive 內容。登入即可解鎖全文,仲有成個 15,000+ 篇、每日更新嘅 AI 動態資料庫。
免費登入/註冊解鎖全文 →Google 或電郵地址即可,即刻解鎖。
