code: def test_high_dim(self):
      config_a = combine_configs(config_1x2_2x2.a, config_2x2_2x2.a)
      config_b = combine_configs(config_1x2_2x2.b, config_2x2_2x2.b)
      config = MatMulConfig(config_a, config_b)
  
      np_a, d_a, np_b, d_b = config.instantiate()
      np_c = np_a @ np_b
      d_c = d_a @ d_b
  
      testing.assert_array_equal(d_c.get(), np_c)
  
imports: import dataclasses
  import math
  import pytest
  import numpy
  import cupy
  from cupy import testing
  from cupyx.distributed.array import _linalg
  from cupyx.distributed import array
  from cupyx.distributed.array import REPLICA, SUM
label_desc: Use image node_id 3 for calling the TestDistributedMatMul obj's underlying member method code with example usage: obj.test_high_dim() without return types