code: def test_blas_linear_shapes(
      tokens: int, in_channels: int, out_channels: int
  ) -> None:
      _blas_linear_helper(
          tokens,
          in_channels,
          out_channels,
          DtypeEnum.fp16,
          ActivationType.IDENTITY,
      )
  
file_path: DeepSpeed/tests/unit/inference/v2/modules/test_blas_linear_module.py
imports: from typing import Optional
  import pytest
  import torch
  from deepspeed.accelerator import get_accelerator
  from deepspeed.inference.v2.inference_utils import ActivationType, DtypeEnum, is_gated
  from deepspeed.inference.v2.modules import ConfigBundle
  from deepspeed.inference.v2.modules.configs import DSLinearConfig
  from deepspeed.inference.v2.modules.interfaces import DSLinearRegistry
  from ...v2.inference_test_utils import allclose
label_desc: Use image node_id 3 for calling a global function with example usage: test_blas_linear_shapes(tokens, in_channels, out_channels) without return types