def test_deterministic_gpu_aamp_ostinato(seed):
m = 50
np.random.seed(seed)
Ts = [np.random.rand(n) for n in [64, 128, 256]]
ref_radius, ref_Ts_idx, ref_subseq_idx = naive.aamp_ostinato(Ts, m)
comp_radius, comp_Ts_idx, comp_subseq_idx = gpu_aamp_ostinato(Ts, m)
npt.assert_almost_equal(ref_radius, comp_radius)
npt.assert_almost_equal(ref_Ts_idx, comp_Ts_idx)
npt.assert_almost_equal(ref_subseq_idx, comp_subseq_idx)
def test_gpu_ostinato():
if not cuda.is_available(): # pragma: no cover
pytest.skip("Skipping Tests No GPUs Available")
m = 50
Ts = [np.random.rand(n) for n in [64, 128, 256]]
ref_radius, ref_Ts_idx, ref_subseq_idx = stumpy.gpu_aamp_ostinato(Ts, m)
comp_radius, comp_Ts_idx, comp_subseq_idx = stumpy.gpu_ostinato(
Ts, m, normalize=False)
npt.assert_almost_equal(ref_radius, comp_radius)
npt.assert_almost_equal(ref_Ts_idx, comp_Ts_idx)
npt.assert_almost_equal(ref_subseq_idx, comp_subseq_idx)