code: def test_positional_embedding(
      seq_lens: List[Tuple[int, int]]
  ) -> None:
      """
      Validate that positional embedding works correctly.
      """
      embed_dim = 4096
      vocab_size = 50304
  
      _ragged_embed_test_helper(
          seq_lens,
          torch.float16,
          torch.int32,
          embed_dim,
          vocab_size,
          pos_embed_size=2048,
      )
  
file_path: DeepSpeed/tests/unit/inference/v2/kernels/ragged_ops/test_ragged_embed.py
imports: from typing import List, Optional, Tuple
  import pytest
  import torch
  from deepspeed.accelerator import get_accelerator
  from deepspeed.inference.v2.kernels.ragged_ops import RaggedEmbeddingKernel
  from ....v2.inference_test_utils import allclose, get_dtypes
  from .ragged_testing_utils import build_batch_and_manager
label_desc: Use image node_id 6 for calling a global function with example usage: test_positional_embedding(seq_lens) without return types