diff --git a/test_fused_direct.py b/test_fused_direct.py index 638eff1..5b87462 100644 --- a/test_fused_direct.py +++ b/test_fused_direct.py @@ -68,14 +68,14 @@ def test_fused_vs_separate(): # Warmup temp = fast_rotate(frame, 45.0) temp = fast_hue_shift(temp, 30.0) - temp = fast_ripple(temp, 15, 10, 2, 0, 960, 540) + temp = fast_ripple(temp, 15, frequency=10, decay=2, phase=0, center_x=960, center_y=540) cp.cuda.Stream.null.synchronize() start = time.time() for i in range(n_frames): temp = fast_rotate(frame, i * 3.6) temp = fast_hue_shift(temp, 30.0) - temp = fast_ripple(temp, 15, 10, 2, i * 0.1, 960, 540) + temp = fast_ripple(temp, 15, frequency=10, decay=2, phase=i * 0.1, center_x=960, center_y=540) cp.cuda.Stream.null.synchronize() separate_time = time.time() - start