def test_chirp(self, time_data_gen, num_samps, f0, t1, f1, method):
cpu_time, gpu_time = time_data_gen(0, 10, num_samps)
cpu_chirp = signal.chirp(cpu_time, f0, t1, f1, method)
gpu_chirp = cp.asnumpy(cusignal.chirp(gpu_time, f0, t1, f1, method))
assert array_equal(cpu_chirp, gpu_chirp)
def test_chirp(num_samps, f0, t1, f1, method):
cpu_time = np.linspace(0, 10, num_samps)
gpu_time = cp.asarray(cpu_time)
cpu_chirp = signal.chirp(cpu_time, f0, t1, f1, method)
gpu_chirp = cp.asnumpy(cusignal.chirp(gpu_time, f0, t1, f1, method))
assert array_equal(cpu_chirp, gpu_chirp)
def gpu_version(self, sig, f0, t1, f1, method):
with cp.cuda.Stream.null:
out = cusignal.chirp(sig, f0, t1, f1, method)
cp.cuda.Stream.null.synchronize()
return out