search

maum-ai / phaseaug

ICASSP 2023 Accepted
https://maum-ai.github.io/phaseaug/
BSD 3-Clause "New" or "Revised" License
188 stars 14 forks source link

Pytorch 2 cannot compile forward_sync #12

Open yzmyyff opened 1 year ago

yzmyyff commented 1 year ago

Hi guys,

Thanks for your work on this project.

I used phaseaug in my vocoder which worked well.

but when I compile the whole model in pytorch 2 and get something wrong.

I minify the codes:

phaseaug = PhaseAug()
compiled_phaseaug = torch.compile(phaseaug.forward_sync)
wav = torch.randn([32, 1, 9600])
compiled_phaseaug(wav, wav)

Do you have any ideas about that?

junjun3518 commented 1 year ago

Sorry, I just tested it on a higher version of torch, without compiling.

I will try it in a month.

junjun3518 commented 1 year ago

@yzmyyff Hi again! I just find this issue that pytorch cannot compile complex valued function. And it seems that pytorch and triton teams do not have time to consider complex in/output. Thus, for now, you should use phaseaug without compile. Thank you for noticing