code: def test_positional_embedding_offset() -> None:
      """
      Validate that positional embedding works correctly with an offset.
      """
      embed_dim = 4096
      vocab_size = 50304
      seq_config = [(1, 877), (619, 0), (213, 372), (1, 45)]
  
      _ragged_embed_test_helper(
          seq_config,
          torch.float16,
          torch.int32,
          embed_dim,
          vocab_size,
          pos_embed_size=2048,
          pos_embed_offset=2,
      )
  
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 7 for calling a global function with example usage: test_positional_embedding_offset() without return types