microsoft / nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
https://nni.readthedocs.io
MIT License
13.88k stars 1.81k forks source link

please support some operator. #5755

Open Sahala08 opened 3 months ago

Sahala08 commented 3 months ago

Describe the bug:

When I use "torch.view_as_complex", the error information is: [2024-03-08 21:48:32] ERROR (nni.compression.pytorch.speedup.jit_translate/MainThread) aten::view_as_complex is not Supported! Please report an issue at https://github.com/microsoft/nni. Thanks~

The "torch.fft.rfft2" and "torch.fft.irfft2" maybe need to be fixed too. Thank you for considering to fix them.

Environment:

Reproduce the problem