窥探Trition的lower(一)

文章预览

作者丨液态黑洞 来源丨https://zhuanlan.zhihu.com/p/695171704 编辑丨GiantPandaCV Triton的存在使得编写针对特定Target的算子的难度大大降低，相比于以往暴露更多硬件细节的编程方法，Triton编程提供了更加简洁的python接口去描述一个算子，并通过multi Level IR(MLIR)(也是多层Dialect)渐进式地lower，从设计上能够更灵活地支持抽象级别的扩展（在最后会对这）。 关于triton的宏观概念和流程，这里就不再展开了。这篇文章的意义在于当你了解了triton的作用（以及一些LLVM的知识），并成功安装后，想要通过实际操作来窥探到Triton内部的运行流程。由于Triton的很多内部函数通过pybind11交叉编译，跳转的过程有时不太直白。因此这篇文章希望通过一个简单的case，来帮助分析和跟踪它每一步在做什么，同时也是自己学习时的记录。此外，这篇文章重点关注编译的流程，因此和ru ………………………………