当前位置：首页 > news >正文

TVM中的Pass两种实现方法？如何选择？

news 来源：原创 2025/5/26 4:06:45

以下是TVM中基于DFPatternCallback和VisitDFPattern两种编写Pass的方法的详细对比与示例总结：

1. 核心概念对比

特性	DFPatternCallback	VisitDFPattern (DFPatternFunctor)
抽象层级	声明式模式匹配	命令式访问者模式
适用场景	简单/中等复杂度的模式匹配	需要精细控制匹配逻辑的复杂场景
代码复杂度	低（自动处理递归匹配）	高（需手动处理嵌套结构）
性能	中等（有模式解析开销）	高（直接操作AST）
类型检查	支持（require_type标志）	需手动实现
典型应用	算子融合、简单图变换	类型推导、复杂条件匹配

2. C++ 实现示例

方法1: DFPatternCallback

// 1. 定义模式匹配回调
class Conv2DReluFuser : public DFPatternCallback {public:Conv2DReluFuser() {auto conv = IsOp("nn.conv2d")({Wildcard(), Wildcard()});pattern_ = IsOp("nn.relu")({conv});require_type_ = true;}Expr Callback(const Expr& pre, const Expr& post,const Map<DFPattern, Array<Expr>>& node_map) override {auto* relu = post.as<CallNode>();auto* conv = relu->args[0].as<CallNode>();return Call(Op::Get("nn.contrib.conv2d_relu"), {conv->args[0], conv->args[1]}, conv->attrs);}
};// 使用方式
Expr Optimize(Expr expr) {Conv2DReluFuser fuser;return RewritePatterns({fuser}, expr);
}

方法2: VisitDFPattern

// 1. 继承DFPatternFunctor
class CustomMatcher : public DFPatternFunctor<bool(const DFPattern&)> {public:bool VisitDFPattern_(const CallPatternNode* op) override {// 检查是否是conv2dif (op->op->name == "nn.conv2d") {return MatchInputPattern(op->args[0]); // 递归检查输入}return false;}bool VisitDFPattern_(const VarPatternNode* op) override {return CheckVarConstraints(op->var);}
};// 使用方式
bool IsConv2D(const DFPattern& pattern) {CustomMatcher matcher;return matcher.VisitDFPattern(pattern);
}

3. Python 实现示例

方法1: DFPatternCallback

# 定义conv2d+relu融合模式
def pattern():conv = is_op("nn.conv2d")(wildcard(), wildcard())return is_op("nn.relu")(conv)# 定义重写函数
def callback(match_expr):conv = match_expr.args[0]return relay.op.nn.contrib.conv2d_relu(conv.args[0], conv.args[1])# 注册Pass
@relay.transform.function_pass(opt_level=2)
class Conv2DReluFuse:def transform_function(self, func, mod, ctx):return relay.rewrite(relay.dataflow_pattern.DFPatternCallback(pattern(), callback, require_type=True), func)

方法2: VisitDFPattern (Python版)

class PatternAnalyzer(relay.ExprVisitor):def __init__(self):self.has_conv2d = Falsedef visit_call(self, call):if call.op.name == "nn.conv2d":self.has_conv2d = Truesuper().visit_call(call)# 使用方式
def contains_conv2d(expr):analyzer = PatternAnalyzer()analyzer.visit(expr)return analyzer.has_conv2d

4. 对比分析

适用场景选择

情况	推荐方法	原因
简单模式匹配	`DFPatternCallback`	声明式语法更简洁，自动处理递归
需要复杂条件判断	`VisitDFPattern`	可完全控制遍历过程，支持自定义逻辑
性能敏感场景	`VisitDFPattern`	避免模式解析开销，直接操作AST
快速原型开发	`DFPatternCallback`	Python接口更友好，代码量少

性能对比

DFPatternCallback：

# 伪代码：内部实现流程
1. 解析Pattern为内部表示
2. 构建模式自动机
3. 线性扫描Expr进行匹配

VisitDFPattern：

// 伪代码：直接访问节点
bool Visit(const Expr& e) {if (e is CallNode) {// 直接处理节点逻辑}
}

扩展性对比

DFPatternCallback扩展：

# 组合多个模式
pattern = is_op("add")(conv_pattern, is_op("mul")(wildcard(), wildcard()))

VisitDFPattern扩展：

// 添加新的节点处理
bool VisitDFPattern_(const NewPatternNode* op) override {// 自定义处理逻辑
}

5. 混合使用建议

在实际TVM Pass开发中，推荐：

先用DFPatternCallback实现基础匹配
对需要特殊处理的节点用VisitDFPattern补充

// C++示例：混合方案
class HybridPass : public DFPatternCallback {public:HybridPass() {pattern_ = ComplexPattern();}Expr Callback(...) override {if (NeedSpecialHandling(post)) {return CustomVisitDFPattern(post);}return DefaultRewrite(post);}
};

两种方法本质上是互补的：

DFPatternCallback 提供声明式的便捷性
VisitDFPattern 提供命令式的灵活性

TVM中的Pass两种实现方法？如何选择？

以下是TVM中基于DFPatternCallback和VisitDFPattern两种编写Pass的方法的详细对比与示例总结： 1. 核心概念对比特性DFPatternCallbackVisitDFPattern (DFPatternFunctor)抽象层级声明式模式匹配命令式访问者模式适用场景简单/中等复杂度的模式匹配需要精细控制匹配…...

编程日记 2025/5/26 4:06:45

JAVA EE_网络原理_UDP与TCP

人海中未遇见时，我将独自前行... ----------陳長生. 1.UDP协议 1.1.UDP协议端格式 UDP（用户数据报协议）是由源端口，目标端口，长度，校验和，数据 5种结构组成。16位是UDP报文中字段的长度&#…...

编程日记 2025/5/26 3:04:50

智能Python开发工具PyCharm v2025.1——AI层级功能重磅升级

JetBrains PyCharm是一种Python IDE，其带有一整套可以帮助用户在使用Python语言开发时提高其效率的工具。此外，该IDE提供了一些高级功能，以用于Django框架下的专业Web开发。立即获取PyCharm v2025.1正式版具体更新内容： PyCh…...

编程日记 2025/4/29 7:12:44

15、项目搭建：绘制城堡蓝图——React 19 工程配置

一、魔法结界初始化 1. 召唤项目骨架 npx create-next-applatest hogwarts-castle --ts --tailwind 核心咒语： • --ts：激活预言水晶球（TypeScript类型安全） • --tailwind：注入飞天扫帚级原子样式（…...

编程日记 2025/5/26 2:53:54

docker搭建swarm集群

环境准备主机名 IP 角色 manger1 192.168.111.47 管理节点 worker1 192.168.111.48 工作节点 worker2 192.168.111.49 工作节点注：三台主机都已经拉去完swarm和…...

编程日记 2025/5/26 4:05:32

普通IT的股票交易成长史--20250428晚

声明：本文章的内容只是自己学习的总结，不构成投资建议。文中观点基本来自yt站Andylee，美股Alpha姐，综合自己的观点得出。感谢他的无私分享。仓位就是生命，绝对不能满仓！！！&#xf…...

编程日记 2025/5/25 11:10:48

【React Native】精通 react native

活到老，学到老。一、基础核心 JavaScript/TypeScript 基础掌握 ES6+ 语法（箭头函数、解构、Promise、async/await）。熟悉 TypeScript（类型系统、接口、泛型）以提高代码质量。React 核心概念组件化开发（函数组件、类组件）。状态管理（useState, useEffect, useContex…...

编程日记 2025/5/26 2:13:48

微信小程序-van-uploader的preview-size

preview-size支持数组格式修改前修改后1、升级微信小程序里面的van版本:2、重新构建npm3、重启微信开发工具修改前引用van组件的上传文件，设置预览图尺寸，刚开始设置的是preview-size“140”，出来的效果就是一个正方形。修改后 1、升级…...

编程日记 2025/5/26 2:34:52

成员方法的详细说明(结合Oracle官方文档)

在Java的对象创建过程中，成员方法的地址并不存储在对象的堆内存中。Java虚拟机的设计说明（包括Oracle的Java虚拟机规范、OpenJDK文档、以及HotSpot的设计文档）都明确区分了对象的实例数据（存储在堆内存中）和类的元数据…...

编程日记 2025/5/26 2:56:50

[蓝桥杯刷题]---模拟法[2]日期问题

题目如下: 题目的意思是： 给出一个日期（格式是yy mm dd，注意年份只有两位数），要找出所有可能的真实日期（合法的yyyy-mm-dd格式）。需要考虑： 年份范围在1960到2059。输入的yy、mm、…...

编程日记 2025/4/29 7:10:35

阿里开源图生动画模型AnimateAnyone2

项目背景近年来，基于扩散模型（diffusion models）的人物图像动画化方法取得了显著进展，例如 Animate Anyone 在生成一致性和泛化性方面表现优异。然而，这些方法在处理人物与环境之间的空间关系和人-物体交互&#xff0…...

编程日记 2025/5/26 2:58:01

02_使用 AES 算法实现文件加密上传至阿里云、解密下载

02_使用 AES 算法实现文件加密上传至阿里云、解密下载一、文件上传下载接口 controller 层 RestController RequestMapping("/api/common/file") Api(tags "公共文件上传") AllArgsConstructor Slf4j public class FileV2Controller {private final Os…...

编程日记 2025/4/29 7:10:33

Linux运维——Vim基础

Vim基础一、移动光标1.1、基础移动1.2、屏幕滚动二、编辑操作2.1、插入模式2.2、删除与修改2.3、复制粘贴三、搜索与替换3.1、搜索3.2、替换 4、分屏与窗口管理4.1、分屏操作4.2、窗口调整五、宏与批量操作六、效率技巧一、移动光标 1.1、基础移动快捷键作用h j k l左/…...

编程日记 2025/4/29 7:10:33

从外卖大战看O2O新趋势：上门私厨平台系统架构设计解析

京东高调进军外卖市场，美团全力防守，两大巨头的竞争让整个行业风起云涌。但在这场外卖大战之外，一个更具潜力的细分市场正在悄然兴起——上门私厨服务。与标准化外卖不同，上门私厨提供的是个性化定制服务。厨师带着新鲜食材上门现…...

编程日记 2025/4/29 7:10:32

【网络编程】ARP协议与主机之间的通信

1. 什么是ARP协议？ ARP（地址解析协议，Address Resolution Protocol） 是一种用于在网络中将IP地址转换为MAC地址的协议。它属于TCP/IP协议栈中的网络层协议，通常在局域网（LAN）环境下使用&#x…...

编程日记 2025/4/29 7:09:30

JVM模型、GC、OOM定位

JVM模型程序计数器程序计数器是一块较小的内存空间，可以看作是当前线程所执行的字节码的行号指示器。字节码解释器工作时通过改变这个计数器的值来选取下一条需要执行的字节码指令，分支、循环、跳转、异常处理、线程恢复等功能都需要依赖这个计数器来…...

编程日记 2025/5/26 3:37:56

什么是数据链路层的CRC检测以及为什么要放到帧尾？

数据链路层在封装过程中添加CRC（循环冗余校验）帧尾，主要目的是为了检测数据传输过程中可能出现的比特错误，确保数据的完整性和可靠性。具体原因如下： 1. 错误检测物理层传输的不可靠性：数据在物理介质&am…...

编程日记 2025/5/26 2:22:38

Electron 入门指南

Electron 入门指南 Electron 是一个使用 JavaScript、HTML 和 CSS 构建跨平台桌面应用的框架。通过 Electron，你可以利用 Web 技术开发出功能强大的桌面应用程序，并且能够运行在 Windows、Mac 和 Linux 系统上。本文将带你从零开始构建一个简单的 Ele…...

编程日记 2025/4/29 7:09:29

目标检测YOLO实战应用案例100讲- 无人机平台下露天目标检测与计数

目录知识储备基于YOLOv8改进的无人机露天目标检测与计数一、环境配置与依赖安装二、核心代码实现（带详细注释） 1. 改进YOLOv8模型定义（添加注意力机制） 2. 无人机视角数据增强（drone_augment.py ） 3. 多目标跟踪与计数（tracking_counter.py ） 4. 完整推理流…...

编程日记 2025/5/23 18:17:23

ArkTS基础实验（二）

任务一：使用模板字符串相关知识，实现多个变量的拼接。同学们可以把自己的姓名、年纪和爱好这三个变量进行拼接。把代码和日志中console.log的打印结果截图保留。预期效果： 任务二：使用状态变量和点击事件相关知识实现计数器案例…...

编程日记 2025/4/29 7:08:25

【计算机视觉】Bayer Pattern与Demosaic算法详解：从传感器原始数据到彩色图像

Bayer Pattern与Demosaic算法详解：从传感器原始数据到彩色图像一、引言在现代数码相机和手机摄像头中，我们能够拍摄到丰富多彩的彩色图像。然而，你可能不知道的是，图像传感器本身并不能直接感知颜色——它们只能感知光的强度。…...

编程日记 2025/5/25 9:12:55

媒体查询使用

一、引言为了确保网页在不同设备上都能提供良好的用户体验，响应式设计变得至关重要。而媒体查询（Media Queries）就是前端开发中实现响应式设计的核心技术之一。二、媒体查询的概念媒体查询是 CSS3 引入的一项强大功能，它允许开…...

编程日记 2025/5/25 10:17:20

deepseek对IBM MQ SSL 证书算法的建议与解答

在IBM MQ配置SSL TLS的命令中，如果参数SSLCIPH使用TLS_RSA_WITH_AES_128_CBC_SHA256，如下所示： DEFINE CHANNEL(QM1.TO.QM2) CHLTYPE(SDR) TRPTYPE(TCP) CONNAME(QM1.MACH.COM) XMITQ(QM2) SSLCIPH(TLS_RSA_WITH_AES_128_CBC_SHA256) DESCR(S…...

编程日记 2025/4/29 7:07:22

服务器文件同步工具有哪些？

服务器文件同步工具的选择取决于你的具体需求（如实时同步、单向/双向同步、跨平台支持、安全性等）。以下是几款主流的服务器文件同步工具推荐，适用于不同场景： 1. 实时同步工具（适合高频率、低延迟需求） rsync 特点：经典增量同步工具，支持本地/远程同步，高效节省带宽。…...

编程日记 2025/4/29 7:07:21

Numpy数组与矩阵——python学习

我前面提到过Numpy函数，但是不够全，在这里我顺便做一些补充。先说明一下我用的是Notebook。一、数组的创建与操作 1、把列表转换为数组 np.array([1,2,3,4,5]) 2、把元组转换为数组 np.array((1,2,3,4,5)) 3、把range对象转换为数组 np.array(rang…...

编程日记 2025/4/29 7:07:21

CasaOS上部署1Panel开源运维面板远程在线访问配置实操指南

文章目录前言1. 添加镜像源2. 部署1Panel3. 本地访问测试4. 安装内网穿透工具5. 配置公网地址6. 配置固定公网地址前言很多时候在尝试远程管理服务器时，常常会遇到各种各样的麻烦，尤其是缺乏公网IP或者路由器设置过于复杂时，更是让人感到…...

编程日记 2025/4/29 7:07:20

深入理解缓存淘汰策略：LRU 与 LFU 算法详解及 Java 实现

一、LRU (Least Recently Used - 最近最少使用) LRU 策略的核心思想是：当缓存空间不足时，优先淘汰最近最长时间未被访问的数据。它基于“时间局部性”原理，即最近被访问的数据，在未来被访问的概率也更高。 LeetCode 146. LRU 缓…...

编程日记 2025/4/29 7:07:19

小智项目架构分析

小智代码架构 .github 这就是github项目上拉下来的一些信息没什么好看的，这跟项目代码无关 .build 编译时生成的文件，没什么可看的，与项目代码无关 .main 主要的代码都在这里面了 .managed_components 这里是小智用到的一些第三方移植…...

编程日记 2025/4/29 7:06:17

基于 SSE 和分块传输的 Uniapp 微信小程序实现流式传输对话

最近的项目是做微信小程序的一个对话框，接入DeepSeek，实现实时对话一个功能。主要用到的技术点为： 1. Server-Sent Events (SSE) 技术： 在请求头中设置了 ‘X-DashScope-SSE’: ‘enable’，启用了SSE协议服务器以事…...

编程日记 2025/5/24 14:15:08

[OS] POSIX C库介绍

POSIX C 库可以理解为 Unix/Linux系统的"标准化工具包"，用一句话概括就是： 👉 它提供了一套跨Unix系统的统一编程接口，让开发者用同一份代码能在不同系统（如Linux、macOS）中运行。核心组成&…...

编程日记 2025/4/29 7:06:15

＜uniapp＞＜插件＞＜UTS＞在uniapp中，创建自己的插件并发布到uni插件市场

前言本专栏是基于uniapp实现手机端各种小功能的程序，并且基于各种通讯协议如http、websocekt等，实现手机端作为客户端（或者是手持机、PDA等），与服务端进行数据通讯的实例开发。发文平台 CSDN 环境配置系统&…...

编程日记 2025/4/29 7:06:13

深度学习前沿探秘：Transformer 模型与多领域应用

技术点目录注意力（Attention）机制详解自然语言处理（NLP）领域的Transformer模型详解计算视觉（CV）领域的Transformer模型详解时间序列建模与预测的大语言模型目标检测算法详解目标检测的大语言模型语义分割的…...

编程日记 2025/4/29 7:05:11

介绍下Nginx的作用与请求转发机制

引言最近笔者在业务中遇到了Nginx轮训策略使用不当导致后端服务的压力增加，从而导致容器CPU资源不足，响应超时的问题； 但由于对Nginx的了解仅限与作为反向代理使用，所以借用GPT工具整理了Nginx的作用以及请求转发机制&#xff…...

编程日记 2025/5/22 2:54:53

Sql刷题日志（day6）

一、笔试 1、insert ignore：在插入数据时忽略主键冲突或其他唯一性约束冲突。如果插入的记录会导致主键冲突（如 actor_id 已存在），该语句不会报错，而是直接忽略插入操作语法： INSERT IGNORE INTO tab…...

编程日记 2025/5/25 22:55:46

Ajax 提交表单与文件上传

目录一、Ajax 提交表单1.1 基本原理1.2 HTML 表单示例1.3 JavaScript 示例（使用 fetch API）二、Ajax 文件上传2.1 基本原理2.2 HTML 文件上传表单示例2.3 JavaScript 示例（使用 fetch API）三、后端处理示例（以 Node.js + Express 为例）3.1 安装依赖3.2 创建服务器文件四…...

编程日记 2025/4/29 7:05:05

【零基础入门】ASP.NET Core快速搭建第一个Web应用

一、为什么选择ASP.NET Core？ 跨平台支持：可在Windows/macOS/Linux系统运行高性能：比传统ASP.NET框架快10倍以上开源生态：活跃的开发者社区和丰富的NuGet包云原生支持：完美适配Docker和Kubernetes部署二、开发…...

编程日记 2025/4/29 7:04:04

盒子模型

1.1看透网页布局的本质 1.2盒子模型的组成部分 css盒子模型本质是一个盒子，封装周围的html元素，它包括边框外边距内边距和实际内容。 padding：盒子与内容间的距离margin：盒子与盒子间的距离 1.3border边框 1.border-width 边…...

编程日记 2025/4/29 7:04:03

厚铜PCB如何兼顾质量与成本？供应商设计规范执行的黄金平衡点

厚铜电路板供应商需要遵循一系列设计规范，以确保所提供的电路板符合行业标准和客户要求。以下是一些需要遵循的设计规范： 1. 电路板尺寸和形状：厚铜电路板供应商需要按照客户提供的规格和要求来设计电路板的尺寸和形状。一般来说，…...

编程日记 2025/4/29 7:04:02

Kafka 配置参数性能调优建议

文章目录 1、生产者调优batch.size（重要）linger.mscompression.typeacks（重要）buffer.memorymax.in.flight.requests.per.connection（重要）message.max.bytes（重要） 2、消费者调优fe…...

编程日记 2025/4/29 7:04:02

Java读Excel：解析阿里云easyExcel导入文件的行号

文章目录引言I 解析阿里云easyExcel导入文件的行号声明解析对象的基类判断Excel解析对象类型是否包含继承某个类 isAssignableFromJava 转换list类型并设置下标到元素对象属性II 封装excel 文件读取excel 文件读取用法文件导入上下文III 参数校验工具类校验参数是否合法 (jaka…...

编程日记 2025/5/24 14:44:06

Vuex持续保存数据页面刷新不丢失 vuex-persistedstate

vuex可以进行全局的状态管理，但刷新后刷新后数据会消失，这是我们不愿意看到的。怎么解决呢，我们可以结合本地存储做到数据状态持久化，但是太麻烦每次都要操作，强烈建议使用插件利用vuex-persistedstate插件. 安装 npm …...

编程日记 2025/4/29 7:03:58

.NET8配置组件

一、组件的概念含义：用于从配置文件中读取配置的组件，叫做配置组件。简单来说，就是从xml、json、yaml、txt等文件中读取配置的组件。主要采用微软提供的 Microsoft.Extensions.Configuration 二、组件的使用 1、json配置读取 (1)、在Dunk.Common.Project.Configuratio…...

编程日记 2025/5/25 2:17:53

加密算法:ed25519和RSA

ed25519 和 RSA 是两种不同的非对称加密算法，常用于 SSH 密钥认证。以下是它们的对比和选择建议： 1. 算法对比特性ed25519RSA (4096-bit)安全性更高（基于椭圆曲线密码学 ECC）高（依赖大数分解难度）密钥长度固定 256 位（公钥/私钥更短）通常 2048/4096 位（公钥较大）性能…...

编程日记 2025/5/24 18:32:10

递归、搜索和回溯算法《递归》

在之前的优选算法当中我们已经学习了一些基本的算法，那么接下来我们就要来学习算法当中的一大重要章节——递归、搜索和回溯算法，其实也就是大家常常听到的dfs、bfs；其实本质就是递归，在学习搜索、回溯等算法的过程当中我们会先来…...

编程日记 2025/4/29 7:02:51

构建“云中”高并发：12306技术改造的系统性启示

作为曾参与12306余票查询系统高并发升级的技术从业者，笔者注意到公众对于12306底层技术常存在认知盲区。为破解这一迷思，特此分享十年前的架构解密文献（该技术之前名叫 gemfire 现已晋升为Apache顶级项目Geode，代码库详见&#xf…...

编程日记 2025/4/29 7:02:51

升级xcode15 报错Error (Xcode): Cycle inside Runner

升级xcode15后报错 Could not build the precompiled application for the device. Error (Xcode): Cycle inside Runner; building could produce unreliable results. This usually can be resolved by moving the shell script phase Thin Binary so that it runs before th…...

编程日记 2025/5/24 16:02:21

1. 核心概念对比

2. C++ 实现示例

方法1: DFPatternCallback

方法2: VisitDFPattern

3. Python 实现示例

方法1: DFPatternCallback

方法2: VisitDFPattern (Python版)

4. 对比分析

适用场景选择

性能对比

扩展性对比

5. 混合使用建议

相关文章：