vla / tests /conftest.py
anhtld's picture
Initial commit: DoVLA-CIL codebase (h=16 breakthrough) (part 2)
20c251e verified
Raw
History Blame Contribute Delete
358 Bytes
from __future__ import annotations
import os
# Login nodes and small CI runners often impose strict process/thread limits. Keep numerical tests
# deterministic and leave production training thread counts to the launcher configuration.
for variable in ("OMP_NUM_THREADS", "OPENBLAS_NUM_THREADS", "MKL_NUM_THREADS"):
os.environ.setdefault(variable, "1")