code: def test_tiled_baddim(in_splits, out_splits):
      dim = 32
      with pytest.raises(RuntimeError):
          l = TiledLinear(
              dim, dim, out_splits=out_splits, in_splits=in_splits
          )
  
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 2 for calling a global function with example usage: test_tiled_baddim(in_splits, out_splits) without return types