Closed
[Documentation|文档反馈]: Ascend 950 支持列表误标 MlaProlog 系列 ACLNN 调用 #6468
KaranocaVeAdd on 18 days agoClosed at 7 days ago
18 days ago Associated pull request
周云飞
18 days ago Comments:
18 days ago Comments:
感谢指出支持列表中不同调用方式的区别。GE 图模式或 Arch35 kernel 存在注册与实现,并不能直接说明对应的公开 ACLNN 接口在 Ascend 950 上可用,支持表应分别描述这些能力。
按您报告的行为,aclnnMlaPrologGetWorkspaceSize、aclnnMlaProlog、aclnnMlaPrologV2WeightNzGetWorkspaceSize 和 aclnnMlaPrologV2WeightNz 在识别到 DAV_3510 或 DAV_9201 时返回 ACLNN_ERR_RUNTIME_ERROR。若这与支持表所对应版本的实现一致,将 mla_prolog 和 mla_prolog_v2 的 op_api 列标为支持,确实会误导调用者选择不可用的接口。
一个合理的调整方向是仅将 docs/zh/ascend950_op_list.md 中这两行的 op_api 标记改为不支持,保留其他列,不将 ACLNN 接口限制扩大为整个算子不支持。当前提供的代码片段尚未覆盖上述平台判断分支、支持表及 GE/Arch35 注册部分,因此这一调整仍需结合这些位置确认适用范围。


周云飞
18 days ago Comments:
18 days ago Comments:
/assign @KaranocaVe


18 days ago Assign KaranocaVe as responsible person
游震
18 days ago Comments:
18 days ago Comments:
/assign @KaranocaVe


18 days ago Modify description of issue
7 days ago Close issue
7 days ago Added label: resolved
[Documentation|文档反馈]: Ascend 950 支持列表误标 MlaProlog 系列 ACLNN 调用
问题
docs/zh/ascend950_op_list.md将mla_prolog和mla_prolog_v2的op_api(ACLNN 调用)标为支持。两个公开 ACLNN 接口在 Ascend 950 上都明确不支持:aclnnMlaPrologGetWorkspaceSize、aclnnMlaProlog、aclnnMlaPrologV2WeightNzGetWorkspaceSize和aclnnMlaPrologV2WeightNz均在识别到DAV_3510或DAV_9201时返回ACLNN_ERR_RUNTIME_ERROR。这是平台支持表与公开接口及实现不一致,调用者会据此选择实际不可用的 ACLNN 接口。GE 图模式和 Arch35 kernel 仍有注册与实现,因此修复只把列表中两行的
op_api标记改为不支持,并保留其他列。关联 PR: https://gitcode.com/cann/ops-transformer/merge_requests/14347