LLVM MLIR怎么用ODS定义自定义操作-C++-PHP中文网

ods是mlir中定义自定义operation的事实标准，需通过.td文件声明式定义；最简transposeop只需命名、输入输出及方言链接，如def transposeop : toy_op { let arguments = (ins f64tensor:$input); let results = (outs f64tensor); }。

llvm mlir怎么用ods定义自定义操作

ODS（Operation Definition Specification）不是“用法选项”，而是 MLIR 中定义自定义 Operation 的事实标准方式。直接手写 C++ 类来继承 mlir::Op 既冗余又易错，ODS 通过 .td 文件声明式地描述操作结构，由 mlir-tblgen 自动产出类型安全、接口完备的 C++ 代码。

怎么写一个最简可用的 `TransposeOp`？

关键不是“全字段堆砌”，而是先跑通骨架：命名、输入、输出、方言链接。其他字段（verifier、parser、printer）可后续补。

以 toy 方言为例，Ops.td 中定义：

def TransposeOp : Toy_Op {
  let summary = "transpose a tensor";
  let arguments = (ins F64Tensor:$input);
  let results = (outs F64Tensor);
}

注意三点：

Toy_Op 必须继承自你方言已定义的基类（如 def Toy_Op : Op），否则 mlir-tblgen 会报 Unknown class 'Toy_Op'
F64Tensor 是你在 Types.td 或同文件中定义的 type alias，不能直接写 tensor —— ODS 不解析具体形状，只认类型名
没有 builders 字段时，生成的 C++ 类仍带默认 builder（如 build(..., Value input)），但若想支持属性（如 {inplace = true}），必须显式加 let arguments = (ins F64Tensor:$input, BoolAttr:$inplace);

`arguments` 和 `results` 怎么区分操作数和属性？

ODS 中所有输入统一归在 arguments，但语义靠修饰符区分：

Aiva.ai

Aiva.ai是一款用于生成多风格原创曲目的 AI 音乐作曲助手。

下载

操作数（operand）：用 Value 类型 + $ 命名，如 F64Tensor:$lhs → 生成 getLhs() 和 getLhsOperand()
属性（attribute）：用 Attr 类型 + $ 命名，如 BoolAttr:$inplace → 生成 getInplace()，且该值在 IR 文本中出现在 {...} 内
混合写法：let arguments = (ins F64Tensor:$input, StrAttr:$layout, I32Attr:$offset); → 对应 MLIR 文本："toy.transpose"(%t) {layout = "nhwc", offset = 42} : ...

漏写 Attr 后缀（比如写成 Str:$layout）会导致 mlir-tblgen 报 Unknown type 'Str'。

为什么 `verifier` 编译失败或不生效？

verifier 字段生成的是 verify() 成员函数，但它**不自动调用** —— 必须确保你的 Dialect 的 initialize() 方法里注册了该 Operation，并且该 Operation 继承链中启用了验证 trait（如 HasCustomVerify）。

常见错误：只写 let verifier = [{ return ::verify(*this); }];，但没在 C++ 端实现 ::verify 函数，或忘了在 OpBase.td 中引入 HasCustomVerify trait
正确做法（在 .td 中）：let traits = ["HasCustomVerify"];，并在对应 Dialect.cpp 实现 LogicalResult verify(TransposeOp op) { ... }
调试技巧：在 verifier 函数里加 llvm::errs() ，然后用 <code>mlir-opt --verify-diagnostics 触发验证，看是否打印

生成的 C++ 代码在哪调用？怎么接入构建系统？

mlir-tblgen 输出两个文件：Ops.h.inc（声明）和 Ops.cpp.inc（实现），它们**不能直接编译**，必须被你的 Dialect.h 和 Dialect.cpp 包含：

Dialect.h 中：#include "Dialect/Ops.h.inc"
Dialect.cpp 中：#include "Dialect/Ops.cpp.inc"
CMakeLists.txt 中需添加 add_mlir_dialect_library 并指定 tblgen_files（如 Ops.td），否则 mlir-tblgen 不会触发