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

Disable GPU persistence by default

Browse Source 
GPU persistence returns CuPy arrays but most primitives expect numpy.
Disable until all primitives support GPU frames.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
giles
2026-02-03 21:24:45 +00:00
parent 6e0ee65e40
commit 2c1728c6ce
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
sexp_effects/primitive_libs/streaming.py
View File

@@ -25,7 +25,8 @@ except ImportError:
CUPY_AVAILABLE = False CUPY_AVAILABLE = False
# GPU persistence mode - output CuPy arrays instead of numpy # GPU persistence mode - output CuPy arrays instead of numpy
GPU_PERSIST = os.environ.get("STREAMING_GPU_PERSIST", "1") == "1" and CUPY_AVAILABLE # Disabled by default until all primitives support GPU frames
GPU_PERSIST = os.environ.get("STREAMING_GPU_PERSIST", "0") == "1" and CUPY_AVAILABLE
# Check for hardware decode support (cached) # Check for hardware decode support (cached)
_HWDEC_AVAILABLE = None _HWDEC_AVAILABLE = None