摘要: 定义 TVM从Halide继承了计算与调度分离的思想,并在其内部重用了部分Halide的调度原语,也引入了一些新的调度原语,用于优化GPU和专用加速器性能。 先举个例子吧: import tvm from tvm import te n = 1024 dtype = "float32" A = te 阅读全文
posted @ 2024-04-14 22:39 牛犁heart 阅读(84) 评论(0) 推荐(0) 编辑
摘要: 定义 移除无用函数,Remove Unused Function,顾名思义,就是删除Module中定义但未用到的函数 当然,它也是一个模块级的优化, 举例子: def get_mod(): mod = tvm.IRModule({}) fn1 = relay.Function([], relay.c 阅读全文
posted @ 2024-04-14 16:13 牛犁heart 阅读(17) 评论(0) 推荐(0) 编辑