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