code: def test_top_2_e_8_gating(n_tokens: int) -> None:
      _top_k_gating_testing_helper(
          n_tokens=n_tokens, n_experts=8, n_top_k=2
      )
  
file_path: deepspeed/tests/unit/inference/v2/kernels/ragged_ops/test_top_k_gating.py
imports: import pytest
  import torch
  import torch.nn.functional
  from deepspeed.accelerator import get_accelerator
  from deepspeed.inference.v2.inference_utils import DtypeEnum
  from deepspeed.inference.v2.kernels.ragged_ops import RaggedTopKGating
  from .ragged_testing_utils import build_simple_batch
  from ...inference_test_utils import allclose
label_desc: Use image node_id 2 for calling a global function with example usage: test_top_2_e_8_gating(n_tokens) without return types