Update device mesh tutorial to use correct API #5063

Job	Run time
pytorch_tutorial_build_worker (5, 15, linux.g5.4xlarge.nvidia.gpu)	24m 27s
pytorch_tutorial_build_worker (1, 15, linux.g5.12xlarge.nvidia.gpu)	15m 5s
pytorch_tutorial_build_worker (10, 15, linux.g5.4xlarge.nvidia.gpu)	16m 42s
pytorch_tutorial_build_worker (2, 15, linux.g5.4xlarge.nvidia.gpu)	25m 6s
pytorch_tutorial_build_worker (8, 15, linux.g5.4xlarge.nvidia.gpu)	38m 10s
pytorch_tutorial_build_worker (11, 15, linux.g5.4xlarge.nvidia.gpu)	27m 59s
pytorch_tutorial_build_worker (7, 15, linux.g5.4xlarge.nvidia.gpu)	21m 59s
pytorch_tutorial_build_worker (3, 15, linux.g5.4xlarge.nvidia.gpu)	25m 58s
pytorch_tutorial_build_worker (4, 15, linux.g5.4xlarge.nvidia.gpu)	33m 53s
pytorch_tutorial_build_worker (13, 15, linux.g5.4xlarge.nvidia.gpu)	15m 36s
pytorch_tutorial_build_worker (14, 15, linux.g5.4xlarge.nvidia.gpu)	28m 46s
pytorch_tutorial_build_worker (6, 15, linux.g5.4xlarge.nvidia.gpu)	28m 24s
pytorch_tutorial_build_worker (9, 15, linux.g5.4xlarge.nvidia.gpu)	20m 37s
pytorch_tutorial_build_worker (12, 15, linux.g5.4xlarge.nvidia.gpu)	16m 42s
pytorch_tutorial_build_worker (15, 15, linux.g5.4xlarge.nvidia.gpu)	19m 25s
pytorch_tutorial_build_manager	22m 25s
	6h 21m 14s

Provide feedback