runtime/kev/__init__.py

"""KEV package defaults."""

import os

# Set before submodules import PyTorch; callers can override this before importing KEV.
os.environ.setdefault("PYTORCH_CUDA_ALLOC_CONF", "expandable_segments:True")