Open ShouyangDong opened 1 month ago
spaces = generate_design_space(
kind="cuda",
mod=self.mod,
target=self.tvm_tgt,
types=None,
sch_rules=actions,
)
score = objective(spaces[0].mod, self.tvm_tgt, self.mod_name, self.inputs)
上面代码中永远采用的是0个mod, 需要放开不同action下面的多个mod,进行mod之间的比较
由于每个动作后得到的sketch有多个代码,比如拆分: 代码1:
代码 2:
这两个代码是等效的,导致最后的搜索空间是正交的, O(sketch) * O(annotation)
减支建议:
ms.schedule_rule.AutoBind()
已经出现1次,后面的action就不会出现该源语