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