code: def test_2d():
      spawn(check_layer_and_operation, 4)
  
file_path: ColossalAI/tests/test_legacy/test_layers/test_2d/test_2d.py
imports: import pytest
  import torch
  from checks_2d.check_layer_2d import check_classifier_given_embed_weight, check_classifier_no_given_weight, check_embed, check_layernorm, check_linear, check_loss, check_patch_embed, check_vocab_parallel_classifier_given_embed_weight, check_vocab_parallel_classifier_no_given_weight, check_vocab_parallel_embed, check_vocab_parallel_loss
  from checks_2d.check_operation_2d 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 4 for calling a global function with example usage: test_2d() without return types