computer_wer returns nan when I use device other than cuda:0

My computer has multiple GPUs, but it seems pytorch-edit-distance only works when using cuda:0.
Whenever I tried using other GPU such as cuda:1 or cuda:2, `computer_wer` returns `nan` even with the same inputs.