[torch.compile] allow tracking forward time

[torch.compile] allow tracking forward time #29067