[Feature] Wrap untraceable functions with torch.fx.wrap (for MMRazor)

[HIT-cwh]

This pr wraps some untraceable functions with torch.fx.wrap so that they can be regarded as leaf nodes in the graph traced by torch.fx.

[CLAassistant]

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
_{You have signed the CLA already but the status is still pending? Let us recheck it.}