3. MLIR核心概念

MLIR（Multi-Level Intermediate Representation）的设计哲学并非重新发明一套全新的编译器中间表示，而是提供一套可扩展、可复用、可组合的基础设施，让不同抽象级别的IR（从接近源代码的高级抽象到接近机器码的低级抽象）能够共存于同一个框架之中。理解MLIR的六个核心概念——Operation、Value、Region、Block、Type、Attribute——是掌握整个MLIR框架的基石。这六个概念并非孤立存在，而是通过严格的层次结构组织在一起，构成了MLIR的“语法骨架”。本章将深入剖析这六个概念的职责、它们之间的嵌套关系，以及在实际Pass开发中的操作方式。

上图展示了MLIR中核心概念的层次关系：一个Module（本身也是一个Operation）包含多个函数（func.func），每个函数内部包含Region，Region内包含Block，Block内包含一系列Operation。Operation之间通过Value传递数据，而Type和Attribute则分别约束Value和Operation的静态属性。下面我们逐一深入剖析。

3.1 Operation：MLIR的基本执行单元

核心原理

Operation（简称Op）是MLIR中最核心的抽象，它表示一个可执行的计算或声明性操作。在MLIR中，几乎一切皆Operation：函数定义、算术运算、控制流、类型转换、模块声明等。每个Operation由以下几部分组成：

• 操作名称（OpName）：形如 arith.addi、func.func，由Dialect前缀和具体操作名组成，确保全局唯一性。
• 操作数（Operands）：零个或多个Value，作为该操作的输入。
• 结果（Results）：零个或多个Value，作为该操作的输出。
• 属性（Attributes）：编译期已知的常量参数，如整数、字符串、数组等。
• 区域（Regions）：零个或多个Region，用于嵌套子IR（如函数体、循环体）。
• 成功回调（Successors）：仅用于具有控制流语义的操作（如分支跳转），指向可能的后续Block。

判断方法或操作流程

在Pass开发中，识别和遍历Operation是基础操作。以下是一个典型的遍历流程：

1. 获取当前Operation的指针（Operation* op）。
2. 通过 op->getName() 获取操作名称，判断是否为目标操作。
3. 通过 op->getOperands() 和 op->getResults() 访问操作数和结果。
4. 通过 op->getAttrs() 遍历属性字典。
5. 通过 op->getRegions() 访问嵌套Region。

// 示例：遍历函数内所有 arith.addi 操作
void processAddiOps(func::FuncOp funcOp) {
  funcOp.walk([&](arith::AddIOp addOp) {
    // 处理每个 addi 操作
    Value lhs = addOp.getLhs();
    Value rhs = addOp.getRhs();
    Value result = addOp.getResult();
    // 打印操作信息
    llvm::errs() << "Found addi: " << lhs << " + " << rhs << " = " << result << "\n";
  });
}

案例拆解：自定义Dialect中的Operation定义

假设我们正在为一种自定义的硬件加速器设计Dialect，需要定义一个 hw.accelerate 操作。该操作接受一个输入张量，并产生一个加速后的输出张量，同时具有一个 mode 属性指定加速模式。

// 使用TableGen定义Operation
def Hw_AccelerateOp : Hw_Op<"accelerate"> {
  let summary = "硬件加速操作";
  let arguments = (ins TensorType:$input, AttrDict:$mode);
  let results = (outs TensorType:$output);
  let assemblyFormat = "$input attr-dict `:` type($input) `->` type($output)";
}

在Pass中，我们可以这样匹配并处理该操作：

// 匹配 hw.accelerate 操作
if (auto accOp = dyn_cast<Hw::AccelerateOp>(op)) {
  // 获取属性
  StringRef mode = accOp.getMode();
  // 获取输入输出
  Value input = accOp.getInput();
  Value output = accOp.getOutput();
  // 根据mode执行不同优化
  if (mode == "fast") { /* 快速模式优化 */ }
  else if (mode == "accurate") { /* 精确模式优化 */ }
}

常见误区与总结复盘

复盘：Operation是MLIR的“动词”，它定义了“做什么”。理解Operation的关键在于区分它的静态部分（名称、属性、类型）和动态部分（操作数、结果、区域）。在编写Pass时，我们通常通过模式匹配（Pattern Match）来定位特定Operation，然后进行变换。

3.2 Value与Type：SSA数据流的核心

核心原理

Value表示SSA（Static Single Assignment）形式中的值。在MLIR中，Value要么由某个Operation的结果产生（OpResult），要么是Block的参数（BlockArgument）。每个Value都关联一个Type，Type描述了该值的形状、元素类型、存储位置等约束。

Type系统是MLIR可扩展性的重要体现。除了内置的 index、i32、f32 等基本类型外，各Dialect可以定义自己的类型。例如：

• tensor<4xf32>：4元素浮点张量
• memref<4x4xf32>：4x4浮点内存缓冲区
• vector<16xf32>：16元素向量
• !quant.quant<i8>：量化类型（来自量化Dialect）

判断方法或操作流程

在Pass中，我们经常需要检查Value的类型并做出决策。以下是一个类型检查的流程：

1. 获取Value对象（Value val）。
2. 调用 val.getType() 获取Type对象。
3. 使用 isa<TensorType>()、dyn_cast<MemRefType>() 等RTTI机制判断具体类型。
4. 对于复合类型，进一步提取元素类型、形状等信息。

// 检查Value是否为4xf32张量
bool is4x32Tensor(Value val) {
  auto tensorType = val.getType().dyn_cast<RankedTensorType>();
  if (!tensorType) return false;
  if (tensorType.getRank() != 1) return false;
  if (tensorType.getDimSize(0) != 4) return false;
  return tensorType.getElementType().isF32();
}

案例拆解：类型转换Pass

考虑一个场景：我们需要将所有的 tensor<4xf32> 转换为 memref<4xf32>，以便后续的内存优化Pass处理。这是一个典型的类型转换Pass：

// 类型转换Pass的核心逻辑
void convertTensorToMemref(Operation* op) {
  op->walk([&](Operation* innerOp) {
    for (Value result : innerOp->getResults()) {
      if (auto tensorType = result.getType().dyn_cast<RankedTensorType>()) {
        // 创建对应的MemRef类型
        MemRefType memrefType = MemRefType::get(
            tensorType.getShape(), tensorType.getElementType());
        // 替换所有使用该Value的地方
        result.replaceAllUsesWith(
            // 插入转换操作（此处省略具体实现）
        );
      }
    }
  });
}

常见误区与总结复盘

复盘：Value和Type构成了MLIR的“名词”部分。Value代表数据流，Type代表数据的静态约束。在Pass开发中，类型检查和转换是最常见的操作之一，熟练掌握Type的RTTI机制和创建方法至关重要。

3.3 Region与Block：结构化控制流的容器

核心原理

Region是Operation内部包含的有序Block列表，它定义了操作的嵌套作用域。Block则是有序Operation列表，代表一个基本块（Basic Block）。Region和Block共同实现了MLIR对结构化控制流（如if-else、for循环）和非结构化控制流（如goto）的表达。

关键特性：

• Region：可以有多个Block，第一个Block是入口块（Entry Block）。Region可以没有Block（空Region），也可以有多个Block（用于表达分支、循环等）。
• Block：可以包含多个Operation，最后一个Operation通常是终结操作（Terminator），如 cf.br、func.return。Block可以有参数（Block Arguments），用于接收来自前驱Block的值。
• 嵌套关系：Operation包含Region，Region包含Block，Block包含Operation。这种递归结构使得MLIR可以表达任意深度的嵌套控制流。

判断方法或操作流程

在Pass中遍历Region和Block是常见需求。以下是一个遍历所有Block的流程：

1. 获取Operation的Region列表（op->getRegions()）。
2. 对每个Region，遍历其Block列表（region->getBlocks()）。
3. 对每个Block，遍历其Operation列表（block->getOperations()）。
4. 检查Block的终结操作（block->getTerminator()）。

// 遍历所有Region和Block
void traverseRegions(Operation* op) {
  for (Region& region : op->getRegions()) {
    for (Block& block : region.getBlocks()) {
      llvm::errs() << "Block with " << block.getNumArguments() 
                << " arguments, " << block.getOperations().size() 
                << " operations\n";
      // 处理终结操作
      Operation* terminator = block.getTerminator();
      if (terminator) {
        llvm::errs() << "  Terminator: " << terminator->getName() << "\n";
      }
    }
  }
}

案例拆解：循环展开Pass

假设我们要实现一个简单的循环展开Pass，将 scf.for 循环展开2次。这需要操作Region和Block：

// 循环展开Pass的核心逻辑
void unrollLoop(scf::ForOp forOp) {
  // 获取循环体Region
  Region& bodyRegion = forOp.getLoopBody();
  Block& bodyBlock = bodyRegion.front();
  
  // 复制循环体操作（展开2次）
  for (int i = 0; i < 2; ++i) {
    // 克隆操作并插入到循环体末尾
    for (Operation& op : bodyBlock.getOperations()) {
      Operation* clone = op.clone();
      bodyBlock.getOperations().insert(++bodyBlock.end(), clone);
    }
  }
  
  // 调整循环边界（此处省略具体实现）
}

常见误区与总结复盘

复盘：Region和Block是MLIR结构化控制流的基石。理解它们的嵌套关系是编写复杂Pass的前提。在实际开发中，我们经常需要同时操作多个Region和Block，因此熟练掌握遍历和修改API至关重要。

3.4 Attribute：编译期常量与元数据

核心原理

Attribute是Operation的编译期常量属性，用于存储编译时已知的信息。与Value不同，Attribute不参与数据流，而是描述Operation的静态特征。常见的Attribute包括：

• IntegerAttr：整数值，如 i32 42
• FloatAttr：浮点值，如 f32 3.14
• StringAttr：字符串，如 "mode"
• ArrayAttr：Attribute数组，如 [1, 2, 3]
• DenseElementsAttr：密集张量数据，如 dense<[1, 2, 3]>
• DictAttr：键值对集合，如 {mode = "fast"}

判断方法或操作流程

在Pass中访问和修改Attribute是常见需求。以下是一个典型的操作流程：

1. 通过 op->getAttr("name") 获取指定名称的Attribute。
2. 使用 dyn_cast<IntegerAttr>() 等RTTI机制判断Attribute类型。
3. 提取Attribute的值（如 attr.getInt()）。
4. 如果需要修改，创建新的Attribute并调用 op->setAttr("name", newAttr)。

// 获取并修改操作的属性
void processAttributes(Operation* op) {
  // 获取现有属性
  if (auto modeAttr = op->getAttrOfType<StringAttr>("mode")) {
    StringRef mode = modeAttr.getValue();
    if (mode == "slow") {
      // 修改属性
      op->setAttr("mode", StringAttr::get(op->getContext(), "fast"));
    }
  }
  
  // 添加新属性
  op->setAttr("optimization_level", IntegerAttr::get(
      IntegerType::get(op->getContext(), 32), 3));
}

案例拆解：基于属性的优化决策

考虑一个场景：我们需要根据操作的 priority 属性决定是否执行某个优化。属性值越高，优先级越高。

// 基于属性的优化决策
void optimizeBasedOnPriority(Operation* op) {
  // 获取priority属性，默认值为0
  int priority = 0;
  if (auto priorityAttr = op->getAttrOfType<IntegerAttr>("priority")) {
    priority = priorityAttr.getInt();
  }
  
  // 根据优先级执行不同优化
  if (priority >= 10) {
    // 高优先级：执行激进优化
    applyAggressiveOptimization(op);
  } else if (priority >= 5) {
    // 中优先级：执行标准优化
    applyStandardOptimization(op);
  } else {
    // 低优先级：仅执行保守优化
    applyConservativeOptimization(op);
  }
}

常见误区与总结复盘

复盘：Attribute是MLIR中表达编译期信息的主要方式。合理使用Attribute可以显著提高Pass的灵活性和可配置性。在开发自定义Dialect时，定义清晰的Attribute体系是良好设计的重要部分。

3.5 综合案例：构建一个简单的类型检查Pass

为了综合运用上述概念，我们设计一个完整的类型检查Pass。该Pass的目标是检查所有 arith.addi 操作的输入类型是否一致，如果不一致则报告错误。

案例背景与约束

在MLIR中，arith.addi 要求两个操作数具有相同的整数类型。但在实际开发中，由于前序Pass的变换，可能会出现类型不匹配的情况。我们需要一个Pass来检测并报告这类错误。

初始现象与目标

假设我们有一个MLIR模块，其中包含以下代码：

func.func @test(%arg0: i32, %arg1: i64) -> i32 {
  %0 = arith.addi %arg0, %arg1 : i32, i64
  return %0 : i32
}

这里 %arg0 是 i32，%arg1 是 i64，但 arith.addi 要求两者类型一致。我们的Pass需要检测出这个错误。

分析判断过程

实现该Pass的关键步骤：

1. 遍历所有 arith.addi 操作。
2. 获取两个操作数的类型。
3. 比较类型是否一致。
4. 如果不一致，报告错误并给出修复建议。

实际操作步骤

// 类型检查Pass实现
struct TypeCheckPass : public PassWrapper<TypeCheckPass, OperationPass<func::FuncOp>> {
  void runOnOperation() override {
    func::FuncOp funcOp = getOperation();
    
    // 遍历所有addi操作
    funcOp.walk([&](arith::AddIOp addOp) {
      Value lhs = addOp.getLhs();
      Value rhs = addOp.getRhs();
      
      // 获取类型
      Type lhsType = lhs.getType();
      Type rhsType = rhs.getType();
      
      // 检查类型一致性
      if (lhsType != rhsType) {
        // 报告错误
        addOp.emitError() << "Type mismatch in addi: "
          << lhsType << " vs " << rhsType;
        
        // 尝试自动修复：将rhs转换为lhs的类型
        OpBuilder builder(addOp);
        builder.setInsertionPoint(addOp);
        Value converted = builder.create<arith::ExtSIOp>(
            addOp.getLoc(), lhsType, rhs);
        addOp.getLhsMutable().assign(converted);
      }
    });
  }
};

结果对比与复盘

运行该Pass后，原始代码中的类型不匹配错误被检测并自动修复：

// 修复前
%0 = arith.addi %arg0, %arg1 : i32, i64

// 修复后
%1 = arith.extsi %arg1 : i64 to i32
%0 = arith.addi %arg0, %1 : i32

3.6 总结与最佳实践

核心概念	角色定位	关键API	常见用途
Operation	执行单元	getName(), getOperands(), getResults()	匹配、变换、删除操作
Value	SSA数据流	getType(), replaceAllUsesWith()	数据依赖分析、值替换
Region	嵌套作用域	getBlocks(), front()	控制流分析、循环变换
Block	基本块	getOperations(), getTerminator()	指令调度、块合并
Type	静态约束	isa<>(), dyn_cast<>()	类型检查、类型转换
Attribute	编译期元数据	getAttr(), setAttr()	配置传递、优化决策