Fix no-grad grad-fn lookup in ZeRO hook counting on PyTorch 2.3 (#7830)#7841
Open
tohtana wants to merge 2 commits intodeepspeedai:masterfrom
Open
Fix no-grad grad-fn lookup in ZeRO hook counting on PyTorch 2.3 (#7830)#7841tohtana wants to merge 2 commits intodeepspeedai:masterfrom
tohtana wants to merge 2 commits intodeepspeedai:masterfrom