已合并
在A5上面跳过test_allgather_into_tensor.py中的两个用例 #38902
在A5上面跳过test_allgather_into_tensor.py中的两个用例 #38902
已合并
xiaoqi-zhou创建于 6月22日
xiaoqi-zhou成员
6月22日

【合入来源】

issue:https://gitcode.com/Ascend/pytorch/issues/2437

【修改方案】

test_all_gather_into_tensor_dist和test_all_gather_into_tensor_uneven_dist调用了torch_npu.npu_format_cast接口,里面调用了aclnnNpuFormatCastCalculateSizeAndFormat算子,在文档上已说明A5不支持ND->NC1HWC0的转换,这两个用例需要在A5上面跳过执行。
aclnnNpuFormatCastCalculateSizeAndFormat接口文档:
srcTensor srcTensor数据格式 dstFormat additionalDtype actualFormat
INT8 ACL_FORMAT_ND(2) ACL_FORMAT_FRACTAL_NZ(29) ACL_INT8(2) ACL_FORMAT_FRACTAL_NZ(29)
INT32 ACL_FORMAT_ND(2) ACL_FORMAT_FRACTAL_NZ(29) ACL_FLOAT16(1)、ACL_BF16(27) ACL_FORMAT_FRACTAL_NZ_C0_16(50)
FLOAT ACL_FORMAT_ND(2) ACL_FORMAT_FRACTAL_NZ(29) ACL_FLOAT16(1)、ACL_BF16(27) ACL_FORMAT_FRACTAL_NZ_C0_16(50)
FLOAT ACL_FORMAT_ND(2) ACL_FORMAT_FRACTAL_NZ(29) ACL_FLOAT8_E4M3FN(36) ACL_FORMAT_FRACTAL_NZ_C0_32(51)
FLOAT16 ACL_FORMAT_ND(2) ACL_FORMAT_FRACTAL_NZ(29) ACL_FLOAT16(1) ACL_FORMAT_FRACTAL_NZ(29)
BFLOAT16 ACL_FORMAT_ND(2) ACL_FORMAT_FRACTAL_NZ(29) ACL_BF16(27) ACL_FORMAT_FRACTAL_NZ(29)
FLOAT8_E4M3FN ACL_FORMAT_ND(2) ACL_FORMAT_FRACTAL_NZ(29) ACL_FLOAT8_E4M3FN(36) ACL_FORMAT_FRACTAL_NZ(29)
FLOAT4_E2M1 ACL_FORMAT_ND(2) ACL_FORMAT_FRACTAL_NZ(29) ACL_FLOAT8_E4M3FN(36) ACL_FORMAT_FRACTAL_NZ(29)
FLOAT4_E2M1 ACL_FORMAT_ND(2) ACL_FORMAT_FRACTAL_NZ(29) ACL_FLOAT4_E2M1(40) ACL_FORMAT_FRACTAL_NZ(29)
FLOAT4_E1M2 ACL_FORMAT_ND(2) ACL_FORMAT_FRACTAL_NZ(29) ACL_FLOAT4_E1M2(41) ACL_FORMAT_FRACTAL_NZ(29)
HIFLOAT8 ACL_FORMAT_ND(2) ACL_FORMAT_FRACTAL_NZ(29) HIFLOAT8(34) ACL_FORMAT_FRACTAL_NZ(29)

【资料变更】

不涉及

【接口变更】

不涉及

【功能验证】

本地测试用例跑通:
image.png
image.png

【CheckList】

PR提交人对以下CheckList自检项进行全量自检,自检通过或不涉及,均修改 [ ] 为 [x]

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 xiaoqi-zhou 的贡献)
Xxiaoqi-zhou成员
6月22日 创建了 pull request,commit 321fcc8e
ascend-robotascend-robot成员
6月22日 添加了label:ascend-cla/yes
ascend-robot
ascend-robot成员
6月22日 评论:

CLA Signature Pass

xiaoqi-zhou, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
ascend-robotascend-robot成员
6月22日 添加了label:needs-issue
此处折叠了42条消息 查看更多
wjlflyer成员
6月22日 评论:

/lgtm

likedislike
renyujin成员
6月22日 评论:

/lgtm
/approve

likedislike
ascend-robotascend-robot成员
6月22日 添加了label:approvedlgtm
ascend-robotascend-robot成员
6月22日 合入了pull request
ascend-robot
ascend-robot成员
6月22日 评论:
流水线 pytorch_gitcode_PR_multiVersion#11046 [ commitID:a588a8ba ] 已完成
likedislike