请教一下这个怎么对应到原始graph,或者有好的方式进一步分析吗?

机器学习PAI使用blade之后出现cuda内存越界访问的问题,用compute-sanitier发现是上面的kernel越界了,请教一下这个怎么对应到原始graph,或者有好的方式进一步分析吗?比如是codegen有问题,还是输入的问题81330d6a34e2f61a9eb5ddf4233778c0.png

展开
收起
真的很搞笑 2023-09-13 13:21:56 56 分享 版权
1 条回答
写回答
取消 提交回答
  • 可以把中间 IR 打印出来看看,这个 kernel 是哪里生成的。IR 的 loc 一般能回溯到原始模型的定位。,此回答整理自钉群“BladeDISC用户支持群”

    2023-09-13 17:44:18
    赞同 展开评论

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

还有其他疑问?
咨询AI助理