Ejemplo n.º 1
0
 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
Ejemplo n.º 2
0
 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)