code: def test_continuation(seq_params: Tuple[int, int]) -> None:
      """
      Test continued generation/prompt processing.
      """
      head_size = 64
      n_heads_q = 32
      n_heads_kv = 32
  
      _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 4 for calling a global function with example usage: test_continuation(seq_params) without return types