load_model 加载 om 模型报错
收藏回复举报
load_model 加载 om 模型报错
t('forum.solved') 已解决
发表于2024-04-02 11:01:54
0 查看

使用 pyacl 加载 om格式模型报错。模型文件是使用了 GE 接口编译出来的离线模型,包含了 HcomAllReduce 通信算子,graph 如下:

cke_3702.png

我们在使用 2 张卡时候,可以正常跑通,但是大于2张卡时候,比如4张卡,就会在 load model的时候报错,报错日志如下:

cke_12153.png

完整的模型和日志见附件。

实验环境:

CANN 版本: 8.0.RC1.alpha001

cpu架构:aarch64

硬件:Atlas 800T A2

我要发帖子