code: def test_gpu_aamp_A_B_join(T_A, T_B):
      m = 3
      for p in [1.0, 2.0, 3.0]:
          ref_mp = naive.aamp(T_B, m, T_B=T_A, p=p, row_wise=True)
          comp_mp = gpu_aamp(T_B, m, T_A, ignore_trivial=False, p=p)
          naive.replace_inf(ref_mp)
          naive.replace_inf(comp_mp)
          npt.assert_almost_equal(ref_mp, comp_mp)
  
imports: from unittest.mock import patch
  import numpy
  import numpy.testing
  import pandas
  from numba import cuda
  from stumpy import config, gpu_aamp
  import naive
  import pytest
label_desc: Use image node_id 4 for calling a global function with example usage: test_gpu_aamp_A_B_join(T_A, T_B) without return types