coop/rose-ash
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Enable JAX compilation for streaming tasks

Browse Source 
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
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
l1/tasks/streaming.py
View File

@@ -354,7 +354,7 @@ def run_stream(
checkpoint = None checkpoint = None
# Create interpreter (pass actor_id for friendly name resolution) # 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 # Set primitive library directory explicitly
interp.primitive_lib_dir = sexp_effects_dir / "primitive_libs" interp.primitive_lib_dir = sexp_effects_dir / "primitive_libs"