code: def test_rms_shapes(rows: int, cols: int, do_residual: bool) -> None:
      _rms_norm_testing_helper(rows, cols, do_residual, DtypeEnum.fp16)
  
file_path: deepspeed/tests/unit/inference/v2/kernels/core_ops/test_rms_norm.py
imports: import pytest
  import torch
  from deepspeed.accelerator import get_accelerator
  from deepspeed.inference.v2.inference_utils import DtypeEnum
  from deepspeed.inference.v2.kernels.core_ops import CUDARMSNorm, CUDARMSPreNorm
  from ....v2.inference_test_utils import get_dtypes, allclose
label_desc: Use image node_id 5 for calling a global function with example usage: test_rms_shapes(rows, cols, do_residual) without return types