Add trace generator

[xuzhao9]

Generate ATen operator traces like https://github.com/meta-pytorch/tritonbench/blob/main/tritonbench/data/input_configs/torchbench_train/BERT_pytorch_training.txt