code: def test_dtype_permutations(
      token_dtype: torch.dtype, embed_dtype: torch.dtype
  ) -> None:
      """
      Validate (on a single problem size) that the kernel support for different data types
      is correct.
      """
      embed_dim = 4096
      vocab_size = 50304
  
      _ragged_embed_test_helper(
          [(256, 0)], embed_dtype, token_dtype, embed_dim, vocab_size
      )
  
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 3 for calling a global function with example usage: test_dtype_permutations(token_dtype, embed_dtype) without return types