[PyTorch] Don't use autograd hook for bwd reduction#781

ksivaman merged 1 commit intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom

ksivaman:fix_hang_for_non_cg_pathksivaman/TransformerEngine-1:fix_hang_for_non_cg_pathCopy head branch name to clipboard