Support torch_musa backend(finished) - Githubissues

yxlllc / ReFlow-VAE-SVC

MIT License

76 stars 12 forks source link

Support torch_musa backend(finished) #2

Closed KakaruHayate closed 3 months ago

KakaruHayate commented 3 months ago

增加对摩尔线程系列GPU对应torch_musa的支持，同时不影响其他功能

MUSA测试环境：

torch2.0.0
torch_musa1.1.0
Intel CORE i3-12100
Moorethreads MTT S70
Ubuntu 20.04 LTS
Python 3.9

CUDA测试环境：

torch2.0.0
python3.8
Windows10
Nvidia RTX A4000

待做：

[x] 预处理部分（unit提取（只考虑hubertsoft）、f0提取（只考虑fcpe？））
[x] 训练
[x] 推理
[x] 声码器（部分算子不支持）