作者JackChung (小锺锺)
看板CompilerDev
标题[分享] 所以,你想用MLIR造语言 0: DEMO
时间Fri Jun 19 11:29:33 2020
庆祝开板来献丑一下, 抛磗引玉.
玩了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://webptt.com/cn.aspx?n=bbs/CompilerDev/M.1592537377.A.3A3.html
1F:推 mshockwave: 未看先推MLIR!小弟一直没有机会玩玩 感谢分享:) 06/20 06:03