code: def test_multi_expert(max_tokens_per_expert: int) -> None:
      """
      Validate for multi-expert GEMM instances that the output is identical to the reference.
      """
      moe_test_helper(
          5120,
          2048,
          64,
          max_tokens_per_expert,
          ActivationType.IDENTITY,
          DtypeEnum.fp16,
      )
  
file_path: DeepSpeed/tests/unit/inference/v2/kernels/cutlass_ops/test_moe_gemm.py
imports: import pytest
  import torch
  from deepspeed.accelerator import get_accelerator
  from deepspeed.inference.v2.inference_utils import ActivationType, DtypeEnum
  from deepspeed.inference.v2.kernels.cutlass_ops import MoEGEMM
  from ....v2.inference_test_utils import allclose
label_desc: Use image node_id 3 for calling a global function with example usage: test_multi_expert(max_tokens_per_expert) without return types