code: def test_extramsa_block(data_args, max_memory):
      spawn(
          run_test,
          1,
          data_args=data_args,
          max_memory=max_memory,
          get_model=get_model,
          get_data=get_data,
      )
  
file_path: colossalai/tests/test_autochunk/test_autochunk_alphafold/test_autochunk_extramsa_block.py
imports: from typing import List, Tuple
  import pytest
  import torch
  import torch.fx
  from test_autochunk_alphafold_utils import run_test
  from colossalai.autochunk.autochunk_codegen import AUTOCHUNK_AVAILABLE
  from colossalai.testing import clear_cache_before_run, parameterize, spawn
label_desc: Use image node_id 3 for calling a global function with example usage: test_extramsa_block(data_args, max_memory) without return types