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)