算子的编译方法、装置、设备及介质

本申请提供了一种算子的编译方法、装置、设备及介质，涉及人工智能技术领域。该算子的编译方法包括：获取待编译的算子源代码，该算子源代码为基于Triton采用领域专用语言编写的代码；将算子源代码转换为Triton方言，进行与硬件无关的优化，该Triton方言为与硬件无关的基于MLIR框架的方言，用于表示算子源代码的计算逻辑；将Triton方言转换为硬件特定方言，该硬件特定方言为与硬件相关的基于MLIR框架的方言，用于表示面向硬件的计算逻辑；根据硬件特定方言，生成采用通用编程语言的源代码；调用目标芯片对应的基于LLVM的编译工具，根据采用通用编程语言的源代码生成目标代码。能够使算子通用于不同芯片架构的芯片。