code: def _check_depth_env_var(depth):
      # check global variable
      env_depth = env.depth_3d
  
      if env_depth:
          assert int(env_depth) == depth, (
              "DEPTH_3D has been set in the current environment and "
              "does not match with the value passed to this initialized"
          )
      else:
          env.depth_3d = depth
  
imports: import math
  import torch.distributed
  from colossalai.legacy.global_variables import tensor_parallel_env
  from colossalai.legacy.registry import DIST_GROUP_INITIALIZER
  from ..parallel_mode import ParallelMode
  from .process_group_initializer import ProcessGroupInitializer
label_desc: Use image node_id 1 for calling a global function with example usage: _check_depth_env_var(depth) without return types