Слияние кода завершено, страница обновится автоматически
Цитата из документации PyTorch:
Если
betaравно 0, тоinputбудет проигнорирован, и nan и inf в нём не будут распространяться.
Но torch_musa распространяет значения nan.
>>> import torch
>>> import torch_musa
>>> dev = torch.device('musa')
>>> inp = torch.full((8, 8, 8), float('nan'), dtype=torch.float16).to(dev)
>>> batch1 = torch.zeros((8, 8, 8), dtype=torch.float16).to(dev)
>>> batch2 = torch.zeros((8, 8, 8), dtype=torch.float16).to(dev)
>>> torch.baddbmm(inp, batch1, batch2, beta=0)
tensor([[[nan, nan, nan, nan, nan, nan, nan, nan],
[nan, nan, nan, nan, nan, nan, nan, nan],
[nan, nan, nan, nan, nan, nan, nan, nan],
[nan, nan, nan, nan, nan, nan, nan, nan],
[nan, nan, nan, nan, nan, nan, nan, nan],
[nan, nan, nan, nan, nan, nan, nan, nan],
[nan, nan, nan, nan, nan, nan, nan, nan],
[nan, nan, nan, nan, nan, nan, nan, nan]],
(дополнительные выводы опущены)
Тестирование выполнено с torch_musa версии 1.1.0+fb1871f.