MLIR中tosa.int_div算子无法识别的问题排查求助

核心原因排查 #

1. TOSA扩展未注册 #

你当前代码注释了mlir::registerAllExtensions(registry);，而tosa.int_div属于TOSA方言的扩展算子范畴。仅调用registerAllDialects只能注册TOSA核心方言，无法覆盖所有扩展算子，必须显式注册扩展才能让MLIR识别这类算子。

2. MLIR构建时未启用TOSA方言 #

如果构建MLIR时未开启TOSA方言的编译开关，即便代码里注册了方言，相关算子定义也不会被编译进程序。LLVM/MLIR默认不会编译所有方言，需要在CMake构建时通过-DMLIR_ENABLE_DIALECTS="TOSA"明确启用TOSA方言支持。

3. LLVM 18.1.8版本不支持该算子 #

虽然LLVM 18是较新版本，但需确认其TOSA方言是否包含tosa.int_div。可以直接查看本地MLIR源码中的mlir/Dialect/Tosa/IR/TosaOps.td文件，搜索int_div是否存在。若文件中无此定义，说明该版本确实不支持，需升级到LLVM 19及以上版本。

修复步骤 #

• 取消注释代码中的mlir::registerAllExtensions(registry);，重新编译cds-opt工具后重试。
• 检查MLIR的CMake构建配置，确保MLIR_ENABLE_DIALECTS包含TOSA，若未添加则重新构建MLIR。
• 若上述步骤无效，查看本地TOSA算子定义文件确认版本支持情况，必要时升级LLVM/MLIR版本。

内容的提问来源于stack exchange，提问作者dsheng.chen