TBE DSL中的auto_schedule能否修改?
收藏回复举报
TBE DSL中的auto_schedule能否修改?
t('forum.solved') 已解决
发表于2023-03-15 17:20:04
0 查看
您好,我在进行TBE DSL算子实验时注意到了auto_schedule,这是一个基于tvm实现的算子自动调度方法。然后我搜索了一下,发现跟它有关的代码是以库包的形式封装在tbe文件夹内(具体为
/home/HwHiAiUser/Ascend/ascend-toolkit/5.0.5.alpha001/arm64-linux/python/site-packages/tbe)

我的问题是:

1)auto_schedule方法能否进行修改,比如对搜索空间进行约束或者修改搜索策略?

2)tbe文件夹内是否包含该方法的实现代码?

我要发帖子