一种基于知识图谱的算子代码生成方法和装置

本发明公开了一种基于知识图谱的算子代码生成方法和装置，包括：根据算子代码库构建代码知识图谱；根据用户输入的检索需求对代码知识图谱进行层次化检索获取参考算子代码树；针对各参考算子代码树生成目标算子代码；对生成的目标算子代码进行筛选获取与检索需求匹配的最终算子代码，其中，最终算子代码应用于AI芯片。通过构建代码知识图谱，并根据用户所输入的检索需求对代码知识图谱进行检索获取参考算子代码树，基于参考代码树生成应用于AI芯片的专有算子代码，从而在代码生成过程中无需专业人员的参与，降低了算子开发成本，提高代码质量，增强算子在特定硬件架构上的适配性。