Add context parallel operator#293

fmassa wants to merge 7 commits intomainmeta-pytorch/autoparallel:mainfrom

fmassa/add_context_parallelmeta-pytorch/autoparallel:fmassa/add_context_parallelCopy head branch name to clipboard