code: def test_spmm(dtype, device):
      row = torch.tensor([0, 0, 1, 2, 2], device=device)
      col = torch.tensor([0, 2, 1, 0, 1], device=device)
      index = torch.stack([row, col], dim=0)
      value = tensor([1, 2, 4, 1, 3], dtype, device)
      x = tensor([[1, 4], [2, 5], [3, 6]], dtype, device)
  
      out = spmm(index, value, 3, 3, x)
      assert out.tolist() == [[7, 16], [8, 20], [7, 19]]
  
imports: from itertools import product
  import pytest
  import torch
  from torch_sparse import spmm
  from torch_sparse.testing import devices, dtypes, tensor
label_desc: Use image node_id 1 for calling a global function with example usage: test_spmm(dtype, device) without return types