Deduplicate split_di_dw_graph by reusing upstream _extract_fwd_bwd_modules#359

xmfan merged 1 commit intomainmeta-pytorch/autoparallel:mainfrom

xmfan/stack/29meta-pytorch/autoparallel:xmfan/stack/29Copy head branch name to clipboard