tongchenkeji 发表于:2023-3-28 10:21:590次点击 已关注取消关注 关注 私信 请问下,bladedisc在确定算子计算调度方案时,是类似用tvm的方式来搜索的吗?[阿里云机器学习PAI] 暂停朗读为您朗读 请问下,bladedisc在确定算子 计算调度方案时,是类似用tvm的方式来搜索的吗 「点点赞赏,手留余香」 赞赏 还没有人赞赏,快来当第一个赞赏的人吧! 海报 机器学习PAI# 人工智能平台 PAI1410# 调度383
爱回答的三好学生AM 2023-11-28 2:58:09 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用户支持群”
凌云CloudAM 2023-11-28 2:58:09 2 bladedisc使用的调度算法和tvm有所不同。 tvm使用的是自动调度算法,而bladedisc使用的是基于遗传算法和启发式算法的离线调度算法。具体来说,bladedisc将计算任务转化为一个离散化的决策问题,然后使用遗传算法和启发式算法搜索最佳的调度方案。 bladedisc github地址
不是的。之前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用户支持群”
bladedisc使用的调度算法和tvm有所不同。
tvm使用的是自动调度算法,而bladedisc使用的是基于遗传算法和启发式算法的离线调度算法。具体来说,bladedisc将计算任务转化为一个离散化的决策问题,然后使用遗传算法和启发式算法搜索最佳的调度方案。
bladedisc github地址