def gpu_version(self, num_samps, a): with cp.cuda.Stream.null: out = cusignal.ricker(num_samps, a) cp.cuda.Stream.null.synchronize() return out
def test_ricker(self, num_samps, a): cpu_window = signal.ricker(num_samps, a) gpu_window = cp.asnumpy(cusignal.ricker(num_samps, a)) assert array_equal(cpu_window, gpu_window)