Fix bug where ISTFT with device="cuda" and move between devices didn't work

qiuqiangkong / torchlibrosa

MIT License

450 stars 45 forks source link

Open iver56 opened 2 years ago

iver56 commented 2 years ago

I've added a test that previously failed*, but passes with the few patches I applied

* E RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

iver56 commented 2 years ago

If this PR is offputting due to the whitespace changes (made by the popular python code formatter black), just let me know, and I can revert those