art-dag/mono - mono - Gitea: Git with a cup of tea

art-dag/mono

Fork 0

Commit Graph

Author	SHA1	Message	Date
giles	07cae101ad	Use JAX for fused pipeline fallback on CPU instead of GPUFrame path All checks were successful Build and Deploy / build-and-deploy (push) Successful in 2m41s Details When CUDA fused kernels aren't available, the fused-pipeline primitive now uses JAX ops (jax_rotate, jax_scale, jax_shift_hue, etc.) instead of falling back to one-by-one CuPy/GPUFrame operations. Legacy GPUFrame path retained as last resort when JAX is also unavailable. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>	2026-02-25 15:35:13 +00:00
giles	4dff4cfafb	Import L1 (celery) as l1/	2026-02-24 23:07:19 +00:00

Author

SHA1

Message

Date

giles

07cae101ad

Use JAX for fused pipeline fallback on CPU instead of GPUFrame path

Build and Deploy / build-and-deploy (push) Successful in 2m41s

Details

When CUDA fused kernels aren't available, the fused-pipeline primitive
now uses JAX ops (jax_rotate, jax_scale, jax_shift_hue, etc.) instead
of falling back to one-by-one CuPy/GPUFrame operations. Legacy GPUFrame
path retained as last resort when JAX is also unavailable.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

2026-02-25 15:35:13 +00:00

giles

4dff4cfafb

Import L1 (celery) as l1/

2026-02-24 23:07:19 +00:00

2 Commits