请问下,bladedisc在确定算子计算调度方案时,是类似用tvm的方式来搜索的吗?[阿里云机器学习PAI]

请问下,bladedisc在确定算子 计算调度方案时,是类似用tvm的方式来搜索的吗

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
=====这是一个广告位,招租中,联系qq 78315851====
1 条回复 A 作者 M 管理员
  1. 不是的。之前bladedisc里主要是调库。 正在做一些cpu/gpu上的计算密集型算子codegen工作,然后加上运行时选择的的策略。没有搜索是因为动态尺寸场景。目前BladeDISC里面基于mlir transform dialect在arm类cpu上做计算密集型子图codegen的一些材料,供参考。 ———

    https://bladedisc.oss-cn-hangzhou.aliyuncs.com/docs/transform-dialect-based-codegen-in-bladedisc.pdf 此回答整理自钉群“BladeDISC用户支持群”

  2. bladedisc使用的调度算法和tvm有所不同。

    tvm使用的是自动调度算法,而bladedisc使用的是基于遗传算法和启发式算法的离线调度算法。具体来说,bladedisc将计算任务转化为一个离散化的决策问题,然后使用遗传算法和启发式算法搜索最佳的调度方案。

    bladedisc github地址