Слияние кода завершено, страница обновится автоматически
docker img: registry.mthreads.com/mcconline/musa-pytorch-dev-public:rc3.1.0-v1.3.0-S4000-py38
musa_toolkit: rc3.1.0
import torch
import torch_musa
input_ = torch.ones(16384, 4096).to(torch.float16).to('musa')
map_ = torch.ones(19030, 4096).to(torch.int64).to('musa')
torch.gather(input_, 0, map_)
Скрипт зависает на torch.gather