[分享] 所以,你想用MLIR造語言 0: DEMO
慶祝開板來獻醜一下, 拋磗引玉.
玩了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
06/20 06:03, 1F