[torch.compile] add a flag to track batchsize statistics

[torch.compile] add a flag to track batchsize statistics #941