code: def set_pipeline_model_parallel_world_size(world_size):
      """Set the pipeline model parallel size"""
      global _MPU_PIPELINE_MODEL_PARALLEL_WORLD_SIZE
      _MPU_PIPELINE_MODEL_PARALLEL_WORLD_SIZE = world_size
  
imports: from typing import Tuple, Optional
  import warnings
  import os
  import torch
  from apex.transformer.log_util import get_transformer_logger
  from apex.transformer._ucc_util import HAS_UCC
label_desc: Use image node_id 28 for calling a global function with example usage: set_pipeline_model_parallel_world_size(world_size) without return types