art-dag/celery
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix fast_ripple signature in test

Browse Source
This commit is contained in:
giles
2026-02-04 09:52:30 +00:00
parent 2d20a6f452
commit b773689814
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test_fused_direct.py
View File

@@ -68,14 +68,14 @@ def test_fused_vs_separate():
# Warmup # Warmup
temp = fast_rotate(frame, 45.0) temp = fast_rotate(frame, 45.0)
temp = fast_hue_shift(temp, 30.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() cp.cuda.Stream.null.synchronize()
start = time.time() start = time.time()
for i in range(n_frames): for i in range(n_frames):
temp = fast_rotate(frame, i * 3.6) temp = fast_rotate(frame, i * 3.6)
temp = fast_hue_shift(temp, 30.0) 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() cp.cuda.Stream.null.synchronize()
separate_time = time.time() - start separate_time = time.time() - start