code: def check_layer():
      check_linear()
      check_layernorm()
      check_embed()
      check_patch_embed()
      check_vocab_parallel_embed()
      check_classifier_no_given_weight()
      check_vocab_parallel_classifier_no_given_weight()
      check_classifier_given_embed_weight()
      check_vocab_parallel_classifier_given_embed_weight()
      check_loss()
      check_vocab_parallel_loss()
  
file_path: colossalai/tests/test_legacy/test_layers/test_2p5d/test_2p5d.py
imports: import pytest
  import torch
  from checks_2p5d.check_layer_2p5d import *
  from checks_2p5d.check_operation_2p5d import check_AB, check_ABT, check_ATB
  from colossalai.legacy.core import global_context
  from colossalai.legacy.initialize import launch
  from colossalai.logging import disable_existing_loggers
  from colossalai.testing import rerun_if_address_is_in_use, spawn
label_desc: Use image node_id 2 for calling a global function with example usage: check_layer() without return types