[PyTorch] Fix stale columnwise data usage#2925

ksivaman wants to merge 1 commit intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom

ksivaman:fix_columnwise_usage_after_evalksivaman/TransformerEngine-1:fix_columnwise_usage_after_evalCopy head branch name to clipboard