code: def test_tiled_forward(in_splits, out_splits, bias, in_f, out_f):
      base = torch.nn.Linear(in_f, out_f, bias=bias)
      test = TiledLinear(
          in_f,
          out_f,
          bias=bias,
          init_linear=copy.deepcopy(base),
          out_splits=out_splits,
          in_splits=in_splits,
      )
  
      inp = torch.rand(in_f)
  
      base_out = base(copy.deepcopy(inp))
      test_out = test(copy.deepcopy(inp))
  
      assert torch.allclose(base_out, test_out, rtol=1e-4)
  
file_path: deepspeed/tests/unit/runtime/zero/test_zero_tiled.py
imports: import copy
  import torch
  from deepspeed.runtime.zero.tiling import TiledLinear, TiledLinearReturnBias
  import pytest
label_desc: Use image node_id 3 for calling a global function with example usage: test_tiled_forward(in_splits, out_splits, bias, in_f, out_f) without return types