def test_unit_impulse(num_samps, idx):
cpu_uimp = signal.unit_impulse(num_samps, idx)
gpu_uimp = cp.asnumpy(cusignal.unit_impulse(num_samps, idx))
assert array_equal(cpu_uimp, gpu_uimp)
def gpu_version(self, num_samps, idx):
with cp.cuda.Stream.null:
out = cusignal.unit_impulse(num_samps, idx)
cp.cuda.Stream.null.synchronize()
return out