[分享] 所以,你想用MLIR造語言 0: DEMO

看板CompilerDev作者 (小鍾鍾)時間4年前 (2020/06/19 11:29), 編輯推噓1(100)
留言1則, 1人參與, 4年前最新討論串1/1
慶祝開板來獻醜一下, 拋磗引玉. 玩了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

06/20 06:03, 4年前 , 1F
未看先推MLIR!小弟一直沒有機會玩玩 感謝分享:)
06/20 06:03, 1F
文章代碼(AID): #1Ux34XEZ (CompilerDev)