Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
foreach_norm大tensor场景下calcBuf越界
910B
import torch tmp = [] for i in range(13): tmp.append(torch.randn([432000256]).float().npu()) torch._foreach_norm(tmp, 2.0)
功能正常
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
Describe the current behavior / 问题描述 (Mandatory / 必填)
foreach_norm大tensor场景下calcBuf越界
Environment / 环境信息 (Mandatory / 必填)
910B
Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)
import torch
tmp = []
for i in range(13):
tmp.append(torch.randn([432000256]).float().npu())
torch._foreach_norm(tmp, 2.0)
Describe the expected behavior / 预期结果 (Mandatory / 必填)
功能正常
Related log / screenshot / 日志 / 截图 (Mandatory / 必填)
Special notes for this issue/备注 (Optional / 选填)