已关闭
[Documentation|文档反馈]: aclnnSparse4to2QuantMatmulWeightNz 的 index 维度说明与实现不一致 #6444
ddssz创建于  19 天前关闭于  18 天前
ddssz
ddssz成员
19 天前 创建

Document Link(文档链接)

aclnnSparse4to2QuantMatmulWeightNz.md

Issues Section(问题文档片段)

index 参数说明将输入和 ViewShape 描述为 4 维,并称 StorageShape 与 ViewShape 相同;维度栏也填写为 4。

同一文档的调用示例使用 2 维 ViewShape:

std::vector<int64_t> indexShape = {n, (k + 7) / 8};

Existing Issues(存在的问题)

参数说明混淆了 ViewShape 和 StorageShape,与示例及接口实现不一致:

Suggested Fix(修复建议)

将 index 的 ViewShape 说明改为 2 维 (n, ceil(k / 8)),同步将维度栏改为 2;明确 StorageShape 由 indexDims 和 indexDimsNum 获取,为 (ceil(k_half / 32), ceil(n / 16), 16, 8),其中 k_half = ceil(k / 8) * 8 / 2。

调用示例中的 2 维 indexShape 与实现一致,无需修改。

关联 PR:https://gitcode.com/cann/ops-nn/merge_requests/11637

likedislike
ddsszddssz成员
19 天前 添加了label:documentation
ddsszddssz成员
19 天前 关联了pull request:docs: 修正Sparse4to2QuantMatmul的index维度说明
ddsszddssz成员
19 天前 修改了issue 的描述
yuning_chenyuning_chen成员
19 天前 将 smdbha 设为负责人
CANN-robotCANN-robot成员
18 天前 关闭了 issue
CANN-robotCANN-robot成员
18 天前 添加了label:resolved