开发者社区 > 大数据与机器学习 > 人工智能平台PAI > 正文

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

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

展开
收起
圆葱猪肉包 2023-03-28 10:21:59 828 0
2 条回答
写回答
取消 提交回答
  • 不是的。之前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用户支持群”

    2023-03-28 11:27:54
    赞同 展开评论 打赏
  • 发表文章、提出问题、分享经验、结交志同道合的朋友

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

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

    bladedisc github地址

    2023-03-28 10:53:54
    赞同 展开评论 打赏

人工智能平台 PAI(Platform for AI,原机器学习平台PAI)是面向开发者和企业的机器学习/深度学习工程平台,提供包含数据标注、模型构建、模型训练、模型部署、推理优化在内的AI开发全链路服务,内置140+种优化算法,具备丰富的行业场景插件,为用户提供低门槛、高性能的云原生AI工程化能力。

热门讨论

热门文章

相关电子书

更多
低代码开发师(初级)实战教程 立即下载
冬季实战营第三期:MySQL数据库进阶实战 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载