看板 CompilerDev 關於我們 聯絡資訊
慶祝開板來獻醜一下, 拋磗引玉. 玩了MLIR好一陣子,總算是有一點點弱弱的成果. 我實作了一個小小的 dialect 來實現 implicit gemm convolution, 然後跑在某冷門平台上. QQ 先上 demo: 編譯過程: https://asciinema.org/a/ZykCGMKdMbwRCqPHTO2Mrs6Q0 編譯加執行過程: https://asciinema.org/a/4zsDYHEZaSP3QQ0GPlfRKoiyR 目前還只是可動的雛型而已,離打敗已商用的基於C++的實作還有一段距離,但是基於 MLIR的實作有不少好處: 1) 各種tensor維度及layout支援都很容易擴充. 2) 近乎無痛嵌入XLA及TensorFlow. 3) 低成本kernel fusion. 4) 多種輸出target : CPU/GPU/FPGA. MLIR可說是提供框架讓你多個願望,一次滿足: 1) 自己定義 domain-specific language. 2) 實作 compiler. 3) 實作 runtime. 4) 實作 debugger QQ. -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 108.226.172.119 (美國) ※ 文章網址: https://www.ptt.cc/bbs/CompilerDev/M.1592537377.A.3A3.html
mshockwave: 未看先推MLIR!小弟一直沒有機會玩玩 感謝分享:) 06/20 06:03