Fix dist autograd context Example block format (#34921)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34921
Test Plan: Imported from OSS
Differential Revision: D20500012
Pulled By: mrshenli
fbshipit-source-id: 6c81123ad347726032c29630d7bf58feb6d8c5fd
