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

Add deterministic debug logging to fused pipeline
Some checks are pending
GPU Worker CI/CD / test (push) Waiting to run
Details
GPU Worker CI/CD / deploy (push) Blocked by required conditions
Details

Browse Source
This commit is contained in:
giles
2026-02-04 12:17:17 +00:00
parent 0a6dc0099b
commit 514ee89cca
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
sexp_effects/primitive_libs/streaming_gpu.py
View File

@@ -873,6 +873,8 @@ def _normalize_effect_dict(effect):
return result return result
_FUSED_CALL_COUNT = 0
def prim_fused_pipeline(img, effects_list, **dynamic_params): def prim_fused_pipeline(img, effects_list, **dynamic_params):
""" """
Apply a fused CUDA kernel pipeline to an image. Apply a fused CUDA kernel pipeline to an image.
@@ -894,6 +896,11 @@ def prim_fused_pipeline(img, effects_list, **dynamic_params):
Supported ops: rotate, zoom, ripple, invert, hue_shift, brightness Supported ops: rotate, zoom, ripple, invert, hue_shift, brightness
""" """
global _FUSED_CALL_COUNT
_FUSED_CALL_COUNT += 1
if _FUSED_CALL_COUNT <= 5 or _FUSED_CALL_COUNT % 100 == 0:
print(f"[FUSED] call #{_FUSED_CALL_COUNT}, effects={len(effects_list)}, params={list(dynamic_params.keys())}", file=sys.stderr)
# Normalize effects list - convert Keyword keys to strings # Normalize effects list - convert Keyword keys to strings
effects_list = [_normalize_effect_dict(e) for e in effects_list] effects_list = [_normalize_effect_dict(e) for e in effects_list]