[main][fix] Add TE's attribute to check if offloading tensor#3896

lhb8125 wants to merge 1 commit intoNVIDIA:mainNVIDIA/Megatron-LM:mainfrom

lhb8125:hongbinl/activation_offloading_mxfp8_fixlhb8125/Megatron-LM:hongbinl/activation_offloading_mxfp8_fixCopy head branch name to clipboard