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

Fix heavy tests to use available primitives

Browse Source
This commit is contained in:
giles
2026-02-04 09:57:29 +00:00
parent f858e25246
commit def62de578
2 changed files with 11 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
test_heavy_fused.sexp
View File

@@ -13,10 +13,9 @@
(require-primitives "image")
(require-primitives "math")
;; Define heavy effects pipeline (5 effects fused into one kernel)
;; Define heavy effects pipeline (4 effects fused into one kernel)
(def heavy-pipeline
[{:op "rotate" :angle 0}
{:op "zoom" :amount 1.0}
{:op "hue_shift" :degrees 30}
{:op "ripple" :amplitude 15 :frequency 10 :decay 2 :phase 0 :center_x 960 :center_y 540}
{:op "brightness" :factor 1.0}])
@@ -31,15 +30,10 @@
base (image:make-image 1920 1080 color)
;; Dynamic parameters
angle1 (* t 30)
angle2 (* t -20)
zoom1 (+ 1.0 (* 0.1 (math:sin (* t 0.5))))
zoom2 (+ 1.0 (* 0.05 (math:cos (* t 0.7))))
phase1 (* t 2)
phase2 (* t 3)]
angle (* t 30)
phase (* t 2)]
;; 10 effects in ONE kernel call!
;; 4 effects in ONE kernel call!
(streaming_gpu:fused-pipeline base heavy-pipeline
:rotate_angle angle1
:zoom_amount zoom1
:ripple_phase phase1))))
:rotate_angle angle
:ripple_phase phase))))