code: def test_matmul(self, config, mode):
      np_a, d_a, np_b, d_b = config.instantiate(mode)
      np_c = np_a @ np_b
      d_c = d_a @ d_b
      testing.assert_array_equal(d_c.get(), np_c, strict=True)
  
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 1 for calling the TestDistributedMatMul obj's underlying member method code with example usage: obj.test_matmul(config, mode) without return types