[torch.compile] fix functionalization

[torch.compile] fix functionalization #15391