TVM 教程:NVIDIA GPU 捲積調優
Apache TVM 是一套端到端深度學習編譯框架,支援 CPU、GPU 及各類 AI 加速晶片。相關教程講解如何在 NVIDIA GPU 上編寫高性能可調模板,配合自動調優器提升捲積運算效率,多數情況可勝過 cuDNN。
調優方法
教程示範為 NVIDIA GPU 編寫高性能、可調整的運算模板,再於模板上運行自動調優器搜索最佳參數組合,在多數情況下性能可超越供應商提供的 cuDNN 庫,展現編譯優化的潛力。
IXThinker 會員內容
呢篇係 Archive 內容。登入即可解鎖全文,仲有成個 15,000+ 篇、每日更新嘅 AI 動態資料庫。
免費登入/註冊解鎖全文 →Google 或電郵地址即可,即刻解鎖。
