Enable JAX compilation for streaming tasks
All checks were successful
Build and Deploy / build-and-deploy (push) Successful in 2m50s

The StreamInterpreter was created without use_jax=True, so the JAX
compiler was never activated for production rendering. Desktop testing
had this enabled but the celery task path did not.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
giles
2026-02-25 15:39:29 +00:00
parent 07cae101ad
commit 4f49985cd5

View File

@@ -354,7 +354,7 @@ def run_stream(
checkpoint = None
# Create interpreter (pass actor_id for friendly name resolution)
interp = StreamInterpreter(str(recipe_path), actor_id=actor_id)
interp = StreamInterpreter(str(recipe_path), actor_id=actor_id, use_jax=True)
# Set primitive library directory explicitly
interp.primitive_lib_dir = sexp_effects_dir / "primitive_libs"