code: def test_parallel_gpu_stump_A_B_join(T_A, T_B):
      device_ids = [device.id for device in cuda.list_devices()]
      if len(T_B) > 10:
          m = 3
          ref_mp = naive.stump(T_B, m, T_B=T_A, row_wise=True)
          comp_mp = gpu_stump(
              T_B,
              m,
              T_A,
              ignore_trivial=False,
              device_id=device_ids,
          )
          naive.replace_inf(ref_mp)
          naive.replace_inf(comp_mp)
          npt.assert_almost_equal(ref_mp, comp_mp)
  
imports: import functools
  from unittest.mock import patch
  import numpy
  import numpy.testing
  import pandas
  from numba import cuda
  from stumpy import config, gpu_stump
  import naive
  import pytest
label_desc: Use image node_id 6 for calling a global function with example usage: test_parallel_gpu_stump_A_B_join(T_A, T_B) without return types