已合并
在A5上面跳过test_allgather_into_tensor.py中的两个用例 #38902
xiaoqi-zhou创建于 6月22日
在A5上面跳过test_allgather_into_tensor.py中的两个用例 #38902
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 xiaoqi-zhou 的贡献)6月22日 创建了 pull request,commit 321fcc8e
ascend-robot
6月22日 评论:
6月22日 评论:
6月22日 添加了label:ascend-cla/yes
ascend-robot
6月22日 评论:
6月22日 评论:
6月22日 添加了label:needs-issue
此处折叠了42条消息 查看更多
6月22日 添加了label:approvedlgtm
6月22日 合入了pull request
ascend-robot
6月22日 评论:
6月22日 评论:
流水线 pytorch_gitcode_PR_multiVersion#11046 [ commitID:a588a8ba ] 已完成


【合入来源】
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)
【资料变更】
不涉及
【接口变更】
不涉及
【功能验证】
本地测试用例跑通:


【CheckList】