def test_square(self, time_data_gen, num_samps, duty):
cpu_time, gpu_time = time_data_gen(0, 10, num_samps)
cpu_pwm = signal.square(cpu_time, duty)
gpu_pwm = cp.asnumpy(cusignal.square(gpu_time, duty))
assert array_equal(cpu_pwm, gpu_pwm)
def test_square(num_samps, duty):
cpu_time = np.linspace(0, 10, num_samps)
gpu_time = cp.asarray(cpu_time)
cpu_pwm = signal.square(cpu_time, duty)
gpu_pwm = cp.asnumpy(cusignal.square(gpu_time, duty))
assert array_equal(cpu_pwm, gpu_pwm)
def gpu_version(self, sig, duty):
with cp.cuda.Stream.null:
out = cusignal.square(sig, duty)
cp.cuda.Stream.null.synchronize()
return out