def test_gausspulse(self, time_data_gen, num_samps, fc):
cpu_time, gpu_time = time_data_gen(0, 10, num_samps)
cpu_pwm = signal.gausspulse(cpu_time, fc, retquad=True, retenv=True)
gpu_pwm = cp.asnumpy(
cusignal.gausspulse(gpu_time, fc, retquad=True, retenv=True))
assert array_equal(cpu_pwm, gpu_pwm)
def gpu_version(self, sig, fc, retquad, retenv):
with cp.cuda.Stream.null:
out = cusignal.gausspulse(sig,
fc,
retquad=retquad,
retenv=retenv)
cp.cuda.Stream.null.synchronize()
return out
def test_gausspulse(num_samps, fc):
cpu_time = np.linspace(0, 10, num_samps)
gpu_time = cp.asarray(cpu_time)
cpu_pwm = signal.gausspulse(cpu_time, fc, retquad=True, retenv=True)
gpu_pwm = cp.asnumpy(
cusignal.gausspulse(gpu_time, fc, retquad=True, retenv=True))
assert array_equal(cpu_pwm, gpu_pwm)