CPU offloading fix: If Data and Transpose is None depend on super Torch tensor class for the shape - #2841

CPU offloading fix: If Data and Transpose is None depend on super Torch tensor class for the shape#2841

vthumbe1503 merged 2 commits into

NVIDIA:mainNVIDIA/TransformerEngine:mainfrom

vthumbe1503:vthumbe/cpu_offloading_bugvthumbe1503/TransformerEngine:vthumbe/cpu_offloading_bugCopy head branch name to clipboard