[PyTorch] Remove is_first_microbatch setting after cudagraph warmup#2715

buptzyb wants to merge 2 commits intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom

buptzyb:robinz/graph_isfirstmicrobatchbuptzyb/TransformerEngine:robinz/graph_isfirstmicrobatchCopy head branch name to clipboard