code: def reference_implementation(hidden_states, weights):
      return hidden_states @ weights.t()
  
file_path: deepspeed/tests/unit/inference/v2/kernels/core_ops/test_blas_linear.py
imports: from typing import Tuple
  import pytest
  import torch
  from deepspeed.accelerator import get_accelerator
  from deepspeed.inference.v2.kernels.core_ops import BlasLibLinear
  from ....v2.inference_test_utils import allclose
label_desc: Use image node_id 1 for calling a global function with example usage: reference_implementation(hidden_states, weights) and returns: unknown