code: def test_gpu_stump_nan_inf_A_B_join(
      T_A, T_B, substitute_A, substitute_B, substitution_locations
  ):
      m = 3
      stop = 16
      T_A_sub = T_A.copy()
      T_B_sub = T_B.copy()[:stop]
  
      for substitution_location_B in substitution_locations:
          for substitution_location_A in substitution_locations:
              T_A_sub[:] = T_A
              T_B_sub[:] = T_B[:stop]
              T_A_sub[substitution_location_A] = substitute_A
              T_B_sub[substitution_location_B] = substitute_B
  
              ref_mp = naive.stump(
                  T_B_sub, m, T_B=T_A_sub, row_wise=True
              )
              comp_mp = gpu_stump(
                  T_B_sub, m, T_A_sub, ignore_trivial=False
              )
              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 13 for calling a global function with example usage: test_gpu_stump_nan_inf_A_B_join(T_A, T_B, substitute_A, substitute_B, substitution_locations) without return types