feat: save forward autocast context in FunctionCtx#185

Chamberlain0w0 wants to merge 2 commits into

masterInfiniTensor/InfiniTrain:masterfrom

feat/autocast_stateInfiniTensor/InfiniTrain:feat/autocast_stateCopy head branch name to clipboard