code: def test_gqa(head_config: Tuple[int, int]) -> None:
      head_size = 128
      n_heads_q = head_config[0]
      n_heads_kv = head_config[1]
  
      seq_params = [(128, 128), (192, 38), (1, 814)]
  
      _blocked_flash_testing_helper(
          head_size, n_heads_q, n_heads_kv, seq_params
      )
  
file_path: DeepSpeed/tests/unit/inference/v2/kernels/ragged_ops/test_blocked_flash.py
imports: import itertools
  from typing import List, Tuple
  import pytest
  import torch
  from deepspeed.accelerator import get_accelerator
  from deepspeed.inference.v2.inference_utils import DtypeEnum
  from deepspeed.inference.v2.kernels.ragged_ops import AtomBuilder, BlockedFlashAttn, get_q_block_size, get_kv_block_size, LinearBlockedKVCopy
  from deepspeed.inference.v2.ragged import split_kv
  from deepspeed.ops.op_builder import RaggedUtilsBuilder
  from .ragged_testing_utils import build_batch_and_manager
  from ....v2.inference_test_utils import allclose
label_desc: Use image node_id 6 for calling a global function with example usage: test_gqa(head_config) without return types