Pytorch 2 cannot compile forward_sync

[yzmyyff]

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]

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

I will try it in a month.

[junjun3518]

@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