[wip] update di/dw partition pass to use new api#309

xmfan wants to merge 1 commit intomainmeta-pytorch/autoparallel:mainfrom

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