In TorchFX we can replace modules after symbolic trace returning a fx.GraphModule, how can we do that for a compiled module from torch.compile?
In TorchFX we can replace modules after symbolic trace returning a fx.GraphModule, how can we do that for a compiled module from torch.compile?
It’s quite similar this notebook should give you more ideas as how this would work Google Colab