1 parent a9fcfbc commit c4a8d63Copy full SHA for c4a8d63
1 file changed
tests/test_gpu_aampdist.py
@@ -44,6 +44,6 @@ def test_gpu_aampdist(T_A, T_B):
44
m = 3
45
for p in [1.0, 2.0, 3.0]:
46
ref_mpdist = naive.aampdist(T_A, T_B, m, p=p)
47
- comp_mpdist = gpu_aampdist(T_A, T_B, m, p=p)
+ cmp_mpdist = gpu_aampdist(T_A, T_B, m, p=p)
48
49
- npt.assert_almost_equal(ref_mpdist, comp_mpdist)
+ npt.assert_allclose(cmp_mpdist, ref_mpdist, atol=1.5e-07)
0 commit comments